| /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 | 70 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {} in ShapeIndex() function
72 ShapeIndex(InputIt start, InputIt end) : indices_(start, end) {} in ShapeIndex() function
172 inline ShapeIndex::ShapeIndex(ShapeIndexView v) 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 const ShapeIndex& /*index*/) { in TryHoistingInvariantInstructionsFromWhileBody()
232 const ShapeIndex& /*index*/) { in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | dynamic_dimension_inference.cc | 962 DimensionConstraint constraint) { in HandleSlice()
1010 hlo, [&](HloInstruction* /*operand*/, ShapeIndex /*index*/, in HandleDynamicUpdateSlice() argument
1029 hlo, [&](HloInstruction* /*operand*/, ShapeIndex /*index*/, in HandleReverse() argument
1046 hlo, [&](HloInstruction* operand, ShapeIndex /*index*/, in HandleGather() argument
1048 HloInstruction* dynamic_size, DimensionConstraint constraint) { in HandleGather()
1112 DimensionConstraint constraint) { in HandleScatter()
1191 HloInstruction* dynamic_size, DimensionConstraint constraint) { in HandleWhile()
|
| D | tuple_points_to_analysis.cc | 52 [&ambiguous](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsAmbiguous()
61 ForEachElement([&](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsDistinct()
81 [&flat_set](const ShapeIndex& /*index*/, const BufferList& buffers) { in CreateFlattenedSet()
99 const ShapeIndex& index) const { in ContainsBufferAtIndex() argument
|
| D | optimize_input_output_buffer_alias_test.cc | 48 [&](const ShapeIndex&, const HloInputOutputAliasConfig::Alias&) { in AliasCount()
|
| D | hlo_input_output_alias_config.cc | 129 [&](const xla::ShapeIndex&, absl::optional<Alias> alias) { in ParameterAliasKind()
|
| D | while_loop_simplifier.cc | 604 subshape, [&](const Shape& s, const ShapeIndex& /*index*/) { in UnflattenTupleInstr()
666 [&](const Shape& s, const ShapeIndex& /*index*/) { in TryFlattenNestedTuples()
|
| D | layout_assignment.cc | 114 [&](const ShapeIndex&, const PointsToSet::BufferList& buffers) { in LayoutConstraints()
140 const PointsToSet::BufferList& buffers) { in GetBufferSet()
|
| D | hlo_memory_scheduler.cc | 123 const PointsToSet::BufferList& buffers) { in ListScheduler()
|
| D | hlo_cost_analysis.cc | 715 [&](const Shape& subshape, const ShapeIndex&) { in HandleAllReduce()
|
| D | hlo_graph_dumper.cc | 1157 [&](const Shape& s, const ShapeIndex&) { in GetInstructionNodeExtraInfo()
|
| D | hlo_verifier.cc | 1246 shape, [&contains_token](const Shape& subshape, const ShapeIndex&) { in ShapeContainsToken()
|
| D | hlo_dataflow_analysis_test.cc | 2674 const ShapeIndex&) { in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | shared_device_buffer.cc | 88 [&](const Shape&, const ShapeIndex&) { consume_buffer(); }); in BufferFromScopedShapedBufferIterator()
136 on_device_shape, [&](const Shape& subshape, const ShapeIndex&) -> Status { in MakeArray()
|
| D | xla.cc | 417 &out, [](Shape* subshape, const ShapeIndex&) { in PYBIND11_MODULE()
|
| D | local_client.cc | 1066 shape, [&](Shape* subshape, const ShapeIndex&) { in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | literal_test_util.cc | 55 const ShapeIndex& /*shape_index*/) { in OnMiscompare()
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 528 [&](const xla::ShapeIndex&, XRTBufferAllocationPtr* buffer) { in MakeTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 266 const ShapeIndex& /*index*/) { in GetIndexTypeForKernel()
|