Searched refs:tensor_shapes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | annotate_max_batch_sizes_test.py | 41 def tensor_shapes(self): member in MaxBatchSizesTestBase
46 return [shape[0] for shape in self.tensor_shapes]
53 input_shapes=[self.tensor_shapes[0]],
54 output_shapes=[self.tensor_shapes[-1]])
103 tensor = array_ops.reshape(tensor, [-1] + self.tensor_shapes[1][1:])
109 tensor = array_ops.reshape(tensor, [-1] + self.tensor_shapes[2][1:])
115 tensor = array_ops.reshape(tensor, [-1] + self.tensor_shapes[3][1:])
129 tensor = array_ops.reshape(tensor, self.tensor_shapes[1])
131 tensor = array_ops.reshape(tensor, self.tensor_shapes[2])
133 tensor = array_ops.reshape(tensor, self.tensor_shapes[3])
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 90 absl::string_view tensor_shapes; in Predict() local
99 tensor_shapes = stat.StrOrRefValue(); in Predict()
105 if (tf_op.type.empty() || tensor_shapes.empty()) { in Predict()
112 for (absl::string_view tensor : ParseTensorShapes(tensor_shapes)) { in Predict()
118 VLOG(1) << tf_op.type << tensor_shapes in Predict()
|
| D | tf_op_utils.cc | 136 absl::string_view tensor_shapes) { in ParseTensorShapes() argument
137 absl::ConsumePrefix(&tensor_shapes, "("); in ParseTensorShapes()
138 absl::ConsumeSuffix(&tensor_shapes, ")"); in ParseTensorShapes()
139 return absl::StrSplit(tensor_shapes, ';'); in ParseTensorShapes()
|
| D | event_span.cc | 131 absl::string_view tensor_shapes) { in ClassifyDeviceCompute() argument
132 if (tensor_shapes.empty()) { in ClassifyDeviceCompute()
141 if (absl::StrContains(tensor_shapes, "half")) in ClassifyDeviceCompute()
178 absl::string_view tensor_shapes) { in ClassifyGpuEvent() argument
188 return ClassifyDeviceCompute(event_name, tensor_shapes); in ClassifyGpuEvent()
|
| D | tf_op_utils.h | 107 absl::string_view tensor_shapes);
|
| D | event_span.h | 227 absl::string_view tensor_shapes);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_step_events.cc | 151 absl::string_view tensor_shapes; in ConvertDeviceTraceXLineToStepEvents() local
163 tensor_shapes = stat.StrOrRefValue(); in ConvertDeviceTraceXLineToStepEvents()
172 EventType event_type = ClassifyGpuEvent(event.Name(), tensor_shapes); in ConvertDeviceTraceXLineToStepEvents()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 177 std::vector<string> tensor_shapes; in ShapeTraceString() local
178 tensor_shapes.reserve(num_inputs); in ShapeTraceString()
181 tensor_shapes.emplace_back(); // Placeholder in ShapeTraceString()
187 tensor_shapes.emplace_back(); // Placeholder in ShapeTraceString()
190 tensor_shapes.emplace_back(strings::StrCat( in ShapeTraceString()
193 return strings::StrCat("(", absl::StrJoin(tensor_shapes, ";"), ")"); in ShapeTraceString()
|