| /external/tensorflow/tensorflow/compiler/xla/service/ |
| gather_expander_test.cc | 97 ShapeUtil::GetTupleElementShape(while_shape, 1)));
101 ShapeUtil::GetTupleElementShape(while_shape, 2)));
105 ShapeUtil::GetTupleElementShape(while_shape, 3)));
|
| hlo_element_type_converter.cc | 70 Shape subshape = ShapeUtil::GetTupleElementShape(shape, i);
91 const Shape& ele_shape = ShapeUtil::GetTupleElementShape(shape, i);
94 const Shape& to_ele_shape = ShapeUtil::GetTupleElementShape(to_shape, i);
|
| allocation_tracker.cc | 162 ShapeUtil::GetTupleElementShape(shaped_buffer->on_host_shape(), i),
163 ShapeUtil::GetTupleElementShape(shaped_buffer->on_device_shape(), i),
|
| hlo_domain_test.cc | 392 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 0), infeed_data,
399 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 1), infeed_data,
646 ShapeUtil::GetTupleElementShape(copy1->shape(), 0), copy1, 0));
649 ShapeUtil::GetTupleElementShape(tuple0->shape(), 1), tuple0, 1));
|
| tuple_simplifier_test.cc | 157 ShapeUtil::GetTupleElementShape(element->shape(), 0), element, 0));
|
| hlo_module_group_metadata.cc | 448 ShapeUtil::GetTupleElementShape(channel.recv_done->shape(), 0);
|
| bfloat16_propagation_test.cc | [all...] |
| hlo_verifier.cc | 709 {ShapeUtil::GetTupleElementShape(recv->shape(), 0),
717 {ShapeUtil::GetTupleElementShape(recv_done->operand(0)->shape(), 0),
[all...] |
| layout_assignment_test.cc | 81 const Layout& actual = ShapeUtil::GetTupleElementShape(shape, i).layout();
246 ShapeUtil::GetTupleElementShape(tuple->shape(), 1), constant1->shape()));
[all...] |
| hlo_cost_analysis.cc | 538 ? ShapeUtil::GetTupleElementShape(fft->operand(0)->shape(), 0)
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_debug.cc | 99 xla::Shape shape0 = xla::ShapeUtil::GetTupleElementShape(padded_shape, 0);
101 xla::ShapeUtil::GetTupleElementShape(padded_shape, 1);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_test.cc | 103 ShapeUtil::GetTupleElementShape(result.shape(), 0),
107 ShapeUtil::GetTupleElementShape(result.shape(), 1),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| tensor_list_utils.cc | 63 xla::ShapeUtil::GetTupleElementShape(list_shape, 0), buffer_shape));
|
| stack_ops.cc | 48 return XLAShapeToTensorShape(xla::ShapeUtil::GetTupleElementShape(shape, 0),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| shape_util.cc | 34 xla::ShapeUtil::GetTupleElementShape(shape, i);
141 xla::ShapeUtil::GetTupleElementShape(input_shape, i);
|
| xla_jit_compiled_cpu_function_test.cc | 128 const xla::Shape& result0 = ShapeUtil::GetTupleElementShape(result, 0);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_transfer_manager.cc | 208 ShapeUtil::GetTupleElementShape(literal_shape, i);
|
| ir_emitter.cc | 333 ShapeUtil::GetTupleElementShape(infeed->shape(), 0)));
343 token_slice, ShapeUtil::GetTupleElementShape(infeed->shape(), 1));
359 ShapeUtil::GetTupleElementShape(data_shape, i);
493 ShapeUtil::GetTupleElementShape(operand_shape, i);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| xrt.cc | 289 ShapeUtil::GetTupleElementShape(tuple_shape, i);
|
| numpy_bridge.cc | 141 PyShapeInfoFromXlaShape(ShapeUtil::GetTupleElementShape(shape, i))
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| tfcompile_test.cc | 586 ShapeUtil::GetTupleElementShape(muladd_result, 0);
589 ShapeUtil::GetTupleElementShape(muladd_result, 1);
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util.h | 473 static const Shape& GetTupleElementShape(const Shape& shape, int64 index);
[all...] |
| shape_util.cc | 343 /* static */ const Shape& ShapeUtil::GetTupleElementShape(const Shape& shape,
866 ShapeUtil::GetTupleElementShape(shape, i), func, index));
[all...] |
| literal_test.cc | [all...] |
| literal_comparison.cc | 193 total += RecursiveElementCount(ShapeUtil::GetTupleElementShape(shape, i));
[all...] |
