| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| H A D | annotate_max_batch_sizes_test.py | 37 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 D | xplane_to_step_events.cc | 73 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 D | cost_utils.cc | 90 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 D | tf_op_utils.cc | 175 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 D | gpu_event_stats.h | 42 absl::string_view tensor_shapes; member
|
| H A D | gpu_event_stats.cc | 41 tensor_shapes = stat.StrOrRefValue(); in GpuEventStats()
|
| H A D | tf_op_utils.h | 137 absl::string_view tensor_shapes);
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | literal_util.cc | 73 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 D | distributed_test.py | 2159 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 D | op_kernel.cc | 218 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 D | test_quantization.py | 230 tensor_shapes = [torch.view_as_real(tensors[0]).shape]
232 tensor_shapes = [tensors[0].shape]
|
| /aosp_15_r20/external/ComputeLibrary/utils/ |
| H A D | GraphUtils.h | 579 … 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 D | special.py | 79 tensor_shapes = ((S, S), ())
82 for shape, n in product(tensor_shapes, ns):
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 2559 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):
|