Searched defs:tensor_shapes (Results 1 – 7 of 7) sorted by relevance
90 absl::string_view tensor_shapes; in Predict() local
136 absl::string_view tensor_shapes) { in ParseTensorShapes()
131 absl::string_view tensor_shapes) { in ClassifyDeviceCompute()178 absl::string_view tensor_shapes) { in ClassifyGpuEvent()
41 def tensor_shapes(self): member in MaxBatchSizesTestBase
73 std::vector<xla::Shape> tensor_shapes(host_tensors.size()); in HostTensorsToBorrowingLiteralTuple() local
151 absl::string_view tensor_shapes; in ConvertDeviceTraceXLineToStepEvents() local
177 std::vector<string> tensor_shapes; in ShapeTraceString() local