| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_support_checker.cc | 31 [&instruction](const Shape& subshape, const ShapeIndex&) { in Run()
|
| D | ir_emitter_unnested.cc | 305 const ShapeIndex& /*index*/) { in GetIndexTypeForKernel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_hlo_support_checker.cc | 31 [&instruction](const Shape& subshape, const ShapeIndex&) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree_test.cc | 65 int_tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestShapeConstructor()
74 [&num_nodes](const ShapeIndex& /*index*/, bool data) { in TestShapeConstructor()
91 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestInitValueConstructor()
99 [&num_nodes](const ShapeIndex& /*index*/, int* data) { in TestInitValueConstructor()
107 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TestInitValueConstructor()
164 [&sum](const ShapeIndex& /*index*/, int data) { sum += data; }); in TEST_F()
176 [](const ShapeIndex& /*index*/, int* data) { *data = 0; }); in TEST_F()
383 t.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) { in TEST_F()
|
| D | shape_util.h | 68 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {} in ShapeIndex() function
70 ShapeIndex(InputIt start, InputIt end) : indices_(start, end) {} in ShapeIndex() function
|
| D | literal_util.cc | 51 &result_shape, [](Shape* subshape, const ShapeIndex&) { in ConvertType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_invariant_code_motion.cc | 223 [&input_size](const Shape& subshape, const ShapeIndex& /*index*/) { in TryHoistingInvariantInstructionsFromWhileBody()
231 [&output_size](const Shape& subshape, const ShapeIndex& /*index*/) { in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | tuple_points_to_analysis.cc | 51 [&ambiguous](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsAmbiguous()
60 ForEachElement([&](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsDistinct()
80 [&flat_set](const ShapeIndex& /*index*/, const BufferList& buffers) { in CreateFlattenedSet()
98 const ShapeIndex& index) const { in ContainsBufferAtIndex() argument
|
| D | optimize_input_output_buffer_alias_test.cc | 53 [&](const ShapeIndex&, const HloInputOutputAliasConfig::Alias&) { in AliasCount()
|
| D | hlo_input_output_alias_config.cc | 124 [&](const xla::ShapeIndex&, absl::optional<Alias> alias) { in ParameterAliasKind()
|
| D | while_loop_simplifier.cc | 603 subshape, [&](const Shape& s, const ShapeIndex& /*index*/) { in UnflattenTupleInstr()
665 [&](const Shape& s, const ShapeIndex& /*index*/) { in TryFlattenNestedTuples()
|
| D | dynamic_dimension_inference.cc | 397 int64 /*operand_index*/, HloInstruction* dynamic_size) { in HandleSlice()
470 HloInstruction* dynamic_size) { in HandleWhile()
|
| D | hlo_cost_analysis.cc | 584 [&](const Shape& subshape, const ShapeIndex&) { in HandleAllReduce()
625 [this](const Shape& subshape, const ShapeIndex& /*shape_index*/) { in HandleFusion()
|
| D | hlo_verifier.cc | 33 shape, [](const Shape& subshape, const ShapeIndex&) -> Status { in VerifyNotSparse()
1056 shape, [&contains_token](const Shape& subshape, const ShapeIndex&) { in ShapeContainsToken()
|
| D | hlo_memory_scheduler.cc | 121 const PointsToSet::BufferList& buffers) { in ListScheduler()
|
| D | layout_assignment.cc | 110 [&](const ShapeIndex&, const PointsToSet::BufferList& buffers) { in LayoutConstraints()
136 const PointsToSet::BufferList& buffers) { in GetBufferSet()
|
| D | hlo_rematerialization.cc | 1225 const ShapeIndex& /*index*/) { in Run()
|
| D | hlo_graph_dumper.cc | 1139 [&](const Shape& s, const ShapeIndex&) { in GetInstructionNodeExtraInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.cc | 332 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) { in ComputeAndCompareLiteralWithStatus()
389 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) { in ComputeAndCompareLiteralWithStatus()
624 [](Shape* subshape, const ShapeIndex&) { in MaybeConvertShapeToBfloat16()
|