/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):
|