Home
last modified time | relevance | path

Searched refs:tensor_shapes (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
H A Dannotate_max_batch_sizes_test.py37 def tensor_shapes(self): member in MaxBatchSizesTestBase
42 return [shape[0] for shape in self.tensor_shapes]
49 input_shapes=[self.tensor_shapes[0]],
50 output_shapes=[self.tensor_shapes[-1]])
99 tensor = array_ops.reshape(tensor, [-1] + self.tensor_shapes[1][1:])
105 tensor = array_ops.reshape(tensor, [-1] + self.tensor_shapes[2][1:])
111 tensor = array_ops.reshape(tensor, [-1] + self.tensor_shapes[3][1:])
125 tensor = array_ops.reshape(tensor, self.tensor_shapes[1])
127 tensor = array_ops.reshape(tensor, self.tensor_shapes[2])
129 tensor = array_ops.reshape(tensor, self.tensor_shapes[3])
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A Dxplane_to_step_events.cc73 absl::string_view tensor_shapes) { in ClassifyGpuCompute() argument
74 if (tensor_shapes.empty()) { in ClassifyGpuCompute()
82 return (absl::StrContains(tensor_shapes, "half")) ? DEVICE_COMPUTE_16 in ClassifyGpuCompute()
88 absl::string_view tensor_shapes) { in ClassifyGpuEvent() argument
99 return ClassifyGpuCompute(event_name, tensor_shapes); in ClassifyGpuEvent()
204 absl::string_view tensor_shapes; in ConvertDeviceTraceXLineToStepEvents() local
216 tensor_shapes = stat.StrOrRefValue(); in ConvertDeviceTraceXLineToStepEvents()
225 EventType event_type = ClassifyGpuEvent(event.Name(), tensor_shapes); in ConvertDeviceTraceXLineToStepEvents()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/
H A Dcost_utils.cc90 absl::string_view tensor_shapes; in Predict() local
98 tensor_shapes = stat.StrOrRefValue(); in Predict()
104 if (tf_op.type.empty() || tensor_shapes.empty()) { in Predict()
111 for (absl::string_view tensor : ParseTensorShapes(tensor_shapes)) { in Predict()
117 VLOG(1) << tf_op.type << tensor_shapes in Predict()
H A Dtf_op_utils.cc175 absl::string_view tensor_shapes) { in ParseTensorShapes() argument
176 absl::ConsumePrefix(&tensor_shapes, "("); in ParseTensorShapes()
177 absl::ConsumeSuffix(&tensor_shapes, ")"); in ParseTensorShapes()
178 return absl::StrSplit(tensor_shapes, ';'); in ParseTensorShapes()
H A Dgpu_event_stats.h42 absl::string_view tensor_shapes; member
H A Dgpu_event_stats.cc41 tensor_shapes = stat.StrOrRefValue(); in GpuEventStats()
H A Dtf_op_utils.h137 absl::string_view tensor_shapes);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dliteral_util.cc73 std::vector<xla::Shape> tensor_shapes(host_tensors.size()); in HostTensorsToBorrowingLiteralTuple() local
80 &tensor_shapes[i])); in HostTensorsToBorrowingLiteralTuple()
84 buf_ptrs, xla::ShapeUtil::MakeTupleShape(tensor_shapes)); in HostTensorsToBorrowingLiteralTuple()
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py2159 tensor_shapes=[tensor.shape],
2421 tensor_shapes=[tensors[0].shape],
2533 tensor_shapes = [tensor_out.shape]
2544 tensor_shapes=tensor_shapes,
2587 tensor_shapes=None, argument
2623 tensor_shapes is not None
2628 tensor_shapes,
2654 tensor_shapes = [torch.view_as_real(tensor).shape]
2656 tensor_shapes = [tensor.shape]
2665 tensor_shapes=tensor_shapes,
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.cc218 std::vector<string> tensor_shapes; in ShapeTraceString() local
219 tensor_shapes.reserve(num_inputs); in ShapeTraceString()
222 tensor_shapes.emplace_back(); // Placeholder in ShapeTraceString()
228 tensor_shapes.emplace_back(); // Placeholder in ShapeTraceString()
231 tensor_shapes.emplace_back(strings::StrCat( in ShapeTraceString()
234 return strings::StrCat("(", absl::StrJoin(tensor_shapes, ";"), ")"); in ShapeTraceString()
/aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/
H A Dtest_quantization.py230 tensor_shapes = [torch.view_as_real(tensors[0]).shape]
232 tensor_shapes = [tensors[0].shape]
/aosp_15_r20/external/ComputeLibrary/utils/
H A DGraphUtils.h579 … std::vector<TensorShape> tensor_shapes,
597 …return std::make_unique<DetectionOutputAccessor>(graph_parameters.labels, tensor_shapes, output_st…
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/
H A Dspecial.py79 tensor_shapes = ((S, S), ())
82 for shape, n in product(tensor_shapes, ns):
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py2559 tensor_shapes = (
2565 for s1, s2, s3 in tensor_shapes:
2571 tensor_shapes = (
2574 for s1, s2, s3 in tensor_shapes:
6878 tensor_shapes = ((S, S), ())
6888 for shape, n in product(tensor_shapes, ns):