Searched defs:tensor_shapes (Results 1 – 6 of 6) sorted by relevance
73 absl::string_view tensor_shapes) { in ClassifyGpuCompute()88 absl::string_view tensor_shapes) { in ClassifyGpuEvent()204 absl::string_view tensor_shapes; in ConvertDeviceTraceXLineToStepEvents() local
90 absl::string_view tensor_shapes; in Predict() local
175 absl::string_view tensor_shapes) { in ParseTensorShapes()
41 def tensor_shapes(self): member in MaxBatchSizesTestBase
73 std::vector<xla::Shape> tensor_shapes(host_tensors.size()); in HostTensorsToBorrowingLiteralTuple() local
182 std::vector<string> tensor_shapes; in ShapeTraceString() local