Searched defs:ShapeIndex (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree_test.cc | 63 int_tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestShapeConstructor()
72 [&num_nodes](const ShapeIndex& /*index*/, bool data) { in TestShapeConstructor()
89 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestInitValueConstructor()
97 [&num_nodes](const ShapeIndex& /*index*/, int* data) { in TestInitValueConstructor()
105 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestInitValueConstructor()
157 [&sum](const ShapeIndex& /*index*/, int data) { sum += data; }); in TEST_F()
169 [&sum](const ShapeIndex& /*index*/, int* data) { *data = 0; }); in TEST_F()
374 t.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TEST_F()
|
| D | shape_util.h | 59 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {} in ShapeIndex() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_support_checker.cc | 32 [&instruction](const Shape& subshape, const ShapeIndex&) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_hlo_support_checker.cc | 32 [&instruction](const Shape& subshape, const ShapeIndex&) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 49 [&ambiguous](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsAmbiguous()
79 [&flat_set](const ShapeIndex& /*index*/, const BufferList& buffers) { in CreateFlattenedSet()
99 const ShapeIndex& index) const { in ContainsBufferAtIndex() argument
|
| D | hlo_cost_analysis.cc | 422 crs->shape(), [&, this](const Shape& subshape, const ShapeIndex&) { in HandleCrossReplicaSum()
457 [this](const Shape& subshape, const ShapeIndex& /*shape_index*/) { in HandleFusion()
|
| D | heap_simulator.cc | 42 const PointsToSet::BufferList& points_to) { in UniqueOperandSourceBuffers()
|
| D | hlo_rematerialization.cc | 1205 const ShapeIndex& /*index*/) { in Run()
|
| D | hlo_graph_dumper.cc | 1073 [&](const Shape& s, const ShapeIndex&) { in GetInstructionNodeExtraInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.cc | 285 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) { in ComputeAndCompareLiteralWithStatus()
337 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) { in ComputeAndCompareLiteralWithStatus()
|
| D | literal_test_util.cc | 113 &result_shape, [](Shape* subshape, const ShapeIndex&) { in ConvertType()
|