Searched refs:TensorByteSize (Results 1 – 2 of 2) sorted by relevance
43 int64 TensorByteSize(const TensorProto& t) { in TensorByteSize() function67 if (TensorByteSize(tp) > kMaxAttrValueTensorByteSize) { in FastTensorProtoHash()105 const int64 lhs_tensor_bytes = TensorByteSize(lhs); in FastAreTensorProtosEqual()106 const int64 rhs_tensor_bytes = TensorByteSize(rhs); in FastAreTensorProtosEqual()
124 size_t TensorByteSize = Spec.getElementCount() * Spec.getElementByteSize(); in writeRawTensorsAsFeatureLists() local141 CurrentTensor += TensorByteSize; in writeRawTensorsAsFeatureLists()