Searched refs:has_tensor (Results 1 – 16 of 16) sorted by relevance
57 def has_tensor(self, tensor_str): function60 CheckpointReader.has_tensor = has_tensor
2677 self.assertTrue(reader.has_tensor("v0"))2678 self.assertTrue(reader.has_tensor("v1"))
138 if (a.has_tensor()) return tensor_hash(a.tensor()); in AttrValueHash()613 if (a.has_tensor() != b.has_tensor()) { in AreAttrValuesEqual()615 } else if (a.has_tensor() && b.has_tensor()) { in AreAttrValuesEqual()
61 if (attr_val.has_tensor()) { in CompressConstants()
157 if (node_value.has_tensor()) { in ByteSwapTensorContent()
263 def has_tensor(obj: object) -> bool: function283 seen_ids[obj_id] = any(has_tensor(v) for v in obj)290 seen_ids[obj_id] = any(has_tensor(v) for v in values)296 seen_ids[obj_id] = any(has_tensor(getattr(obj, v)) for v in obj._fields)307 if has_tensor(value):
38 if (attr_val.has_tensor()) { in NormalizeNode()
86 if not reader.has_tensor(tensor_name):
797 bool has_tensor = false; in PrepareExternal() local801 has_tensor = true; in PrepareExternal()807 has_tensor = true; in PrepareExternal()810 if (has_tensor) { in PrepareExternal()
725 bool has_tensor = false; in PrepareExternal() local729 has_tensor = true; in PrepareExternal()735 has_tensor = true; in PrepareExternal()738 if (has_tensor) { in PrepareExternal()
88 ASSERT_TRUE(iter->second.has_tensor()); in TEST_F()
560 if (!iter->second.has_tensor() || in GrapplerItemFromMetaGraphDef()
88 if (node_value.has_tensor()) { in ValidateNode()
346 if (attr.second.has_tensor() && in IsScalarConst()
113 bool has_tensor = preMulSupplement->tensor_factor.defined(); in unpackPreMulSum() local114 auto residence = has_tensor ? ncclScalarDevice : ncclScalarHostImmediate; in unpackPreMulSum()115 const T* ptr_factor = has_tensor in unpackPreMulSum()123 /*scalar=*/has_tensor ? const_cast<T*>(ptr_factor) : &scalar_factor, in unpackPreMulSum()
2980 if (!value_it->second.has_tensor()) { in ExtractConstTensorFromGraph()