| /external/tensorflow/tensorflow/compiler/xla/service/ |
| logical_buffer.h | 66 // ShapeUtil::GetSubshape(). Examples:
108 // defined. Index used defined as in ShapeUtil::GetSubshape()
131 return ShapeUtil::GetSubshape(instruction_->shape(), index_);
|
| liveness_util.cc | 190 ShapeUtil::GetSubshape(operand->shape(), operand_index);
192 ShapeUtil::GetSubshape(user->shape(), user_index);
283 ShapeUtil::GetSubshape(operand->shape(), operand_index);
285 ShapeUtil::GetSubshape(user->shape(), user_index);
|
| buffer_liveness_test.cc | 70 ShapeUtil::Compatible(ShapeUtil::GetSubshape(a->shape(), index),
71 ShapeUtil::GetSubshape(b->shape(), index)));
488 ShapeUtil::GetSubshape(tuple_param0->shape(), {0});
500 ShapeUtil::GetSubshape(tuple_param0->shape(), {1});
552 ShapeUtil::GetSubshape(tuple_param0->shape(), {0});
564 ShapeUtil::GetSubshape(tuple_param0->shape(), {1});
[all...] |
| hlo_value.cc | 43 return ShapeUtil::GetSubshape(instruction->shape(), index);
|
| transfer_manager.cc | 203 const Shape& subshape = ShapeUtil::GetSubshape(on_device_shape, index);
|
| copy_insertion_test.cc | 346 ShapeUtil::GetSubshape(param->shape(), {0}), param, 0));
383 ShapeUtil::GetSubshape(select->shape(), {0}), select, 0));
728 ShapeUtil::GetSubshape(xla_while->shape(), {1}), xla_while, 1));
733 ShapeUtil::GetSubshape(xla_while->shape(), {0}), xla_while, 0));
[all...] |
| layout_assignment.cc | 78 ShapeUtil::GetSubshape(instruction->shape(), {i}), instruction,
85 ShapeUtil::GetSubshape(shape_with_layout, {i}), gte));
[all...] |
| bfloat16_normalization_test.cc | 245 EXPECT_EQ(ShapeUtil::GetSubshape(crs->shape(), {1}).element_type(), F32);
|
| bfloat16_normalization.cc | 160 output_types[i] = ShapeUtil::GetSubshape(crs->shape(), {i}).element_type();
|
| hlo_sharding.cc | 187 ShapeUtil::GetSubshape(shape, index_to_sharding.first), num_devices);
|
| layout_assignment_test.cc | 310 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::GetSubshape(result_shape, {0}),
312 EXPECT_TRUE(ShapeUtil::Equal(ShapeUtil::GetSubshape(result_shape, {1}),
[all...] |
| buffer_assignment_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util_test.cc | 376 TEST(ShapeUtilTest, GetSubshape) {
380 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(array_shape, {})));
388 ShapeUtil::Equal(tuple_shape, ShapeUtil::GetSubshape(tuple_shape, {})));
390 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {0})));
392 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {1})));
394 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {2})));
403 ShapeUtil::GetSubshape(nested_tuple_shape, {})));
405 array_shape, ShapeUtil::GetSubshape(nested_tuple_shape, {0})));
408 ShapeUtil::GetSubshape(nested_tuple_shape, {1})));
411 ShapeUtil::GetSubshape(nested_tuple_shape, {2, 0})))
[all...] |
| literal_util.cc | 96 piece.set_subshape(&ShapeUtil::GetSubshape(shape_, index));
135 piece.set_subshape(&ShapeUtil::GetSubshape(shape_, index));
153 piece.set_subshape(&ShapeUtil::GetSubshape(shape_, index));
255 elements.push_back(Literal(ShapeUtil::GetSubshape(shape(), {i}),
357 ShapeUtil::GetSubshape(shape(), dest_shape_index);
359 ShapeUtil::GetSubshape(src_literal.shape(), src_shape_index);
401 ShapeUtil::GetSubshape(shape(), dest_shape_index);
[all...] |
| shape_tree.h | 139 // given index (see ShapeUtil::GetSubshape for how indexes are defined).
251 // index : the index of the element in the shape. See ShapeUtil::GetSubshape
297 // ShapeUtil::GetSubshape).
626 ShapeUtil::GetSubshape(shape(), target_base_index),
627 ShapeUtil::GetSubshape(other.shape(), source_base_index)));
|
| shape_util.h | 38 // ShapeUtil::GetSubshape and other interfaces. Shapes are recursive data
433 // GetSubshape and GetMutableSubshape return a particular nested Shape within
435 static const Shape& GetSubshape(const Shape& shape, ShapeIndexView index);
|
| literal_util.h | 612 return ShapeUtil::ElementsIn(ShapeUtil::GetSubshape(shape(), index));
[all...] |
| shape_util.cc | 816 /* static */ const Shape& ShapeUtil::GetSubshape(const Shape& shape,
839 return !IsTuple(GetSubshape(shape, index));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| hlo_to_ir_bindings.cc | 143 ShapeUtil::GetSubshape(hlo.shape(), shape_index), module_);
204 ShapeUtil::GetSubshape(hlo.shape(), shape_index));
|
| ir_emitter_unnested.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_transfer_manager.cc | 122 const Shape& tuple_element_shape = ShapeUtil::GetSubshape(shape, {i});
|
| ir_emitter.cc | [all...] |
