Merge pull request #42011 from amturati:fixing_zerosLike
PiperOrigin-RevId: 324712864 Change-Id: I020caf38ef31943a538a806523f0e50e8de20fc7
This commit is contained in:
commit
5666240e19
@ -101,7 +101,7 @@ AbstractTensorHandle* TapeTensor::ZerosLike() const {
|
|||||||
}
|
}
|
||||||
if (isa<tracing::TracingOperation>(op.get())) {
|
if (isa<tracing::TracingOperation>(op.get())) {
|
||||||
s = dyn_cast<tracing::TracingOperation>(op.get())->SetOpName(
|
s = dyn_cast<tracing::TracingOperation>(op.get())->SetOpName(
|
||||||
absl::StrCat("OnesLike", ToId(handle_)).c_str());
|
absl::StrCat("ZerosLike", ToId(handle_)).c_str());
|
||||||
if (!s.ok()) {
|
if (!s.ok()) {
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user