Searched defs:tensor_shapes (Results 1 – 8 of 8) 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
42 absl::string_view tensor_shapes; member
175 absl::string_view tensor_shapes) { in ParseTensorShapes()
37 def tensor_shapes(self): member in MaxBatchSizesTestBase
73 std::vector<xla::Shape> tensor_shapes(host_tensors.size()); in HostTensorsToBorrowingLiteralTuple() local
218 std::vector<string> tensor_shapes; in ShapeTraceString() local
2587 tensor_shapes=None, argument