| /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 | 71 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {} in ShapeIndex() function
73 ShapeIndex(InputIt start, InputIt end) : indices_(start, end) {} in ShapeIndex() function
173 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 | tuple_points_to_analysis.cc | 54 [&ambiguous](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsAmbiguous()
63 ForEachElement([&](const ShapeIndex& /*index*/, const BufferList& points_to) { in IsDistinct()
83 [&flat_set](const ShapeIndex& /*index*/, const BufferList& buffers) { in CreateFlattenedSet()
101 const ShapeIndex& index) const { in ContainsBufferAtIndex() argument
|
| D | dynamic_dimension_inference.cc | 1119 hlo, [&](HloInstruction*, ShapeIndex /*index*/, int64 dimension, in HandleDynamicSlice() argument
1143 [&](HloInstruction* /*operand*/, ShapeIndex /*index*/, int64 dimension, in HandleDynamicUpdateSlice() argument
1144 int64 operand_index, HloInstruction* dynamic_size) { in HandleDynamicUpdateSlice()
1190 hlo, [&](HloInstruction* operand, ShapeIndex /*index*/, in HandleGather() argument
1192 HloInstruction* dynamic_size) { in HandleGather()
1283 HloInstruction* dynamic_size) -> Status { in HandleConditional()
1408 int64 operand_index, HloInstruction* operand_dynamic_size) { in HandleScatter()
|
| D | optimize_input_output_buffer_alias_test.cc | 48 [&](const ShapeIndex&, const HloInputOutputAliasConfig::Alias&) { in AliasCount()
|
| D | while_loop_expensive_invariant_code_motion.cc | 282 const ShapeIndex& /*index*/) { in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | hlo_input_output_alias_config.cc | 128 [&](const xla::ShapeIndex&, absl::optional<Alias> alias) { in ParameterMustAlias()
|
| D | while_loop_simplifier.cc | 815 subshape, [&](const Shape& s, const ShapeIndex& /*index*/) { in UnflattenTupleInstr()
877 [&](const Shape& s, const ShapeIndex& /*index*/) { in TryFlattenNestedTuples()
|
| D | hlo_memory_scheduler.cc | 124 const PointsToSet::BufferList& buffers) { in ListScheduler()
|
| D | layout_assignment.cc | 115 [&](const ShapeIndex&, const PointsToSet::BufferList& buffers) { in LayoutConstraints()
141 const PointsToSet::BufferList& buffers) { in GetBufferSet()
|
| D | hlo_instructions.h | 1501 std::vector<std::pair<ShapeIndex, std::pair<int64, ShapeIndex>>> in set_output_to_operand_aliasing()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer_test.cc | 35 [&](const Shape& subshape, const ShapeIndex&) -> Status { in MakeArray()
|
| D | tracked_device_buffer.cc | 107 shaped_buffer->on_device_shape(), [&](const Shape&, const ShapeIndex&) { in FromScopedShapedBuffer()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_executable_info.proto | 39 message ShapeIndex { message
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | literal_test_util.cc | 53 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/xla/tools/ |
| D | run_hlo_module.cc | 79 const ShapeIndex& /*shape_index*/) { in OnMiscompare()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 143 &shape_with_layout, [](Shape* subshape, const ShapeIndex&) { in PYBIND11_MODULE()
|
| D | outfeed_receiver.cc | 423 &shape_with_layout, [](Shape* subshape, const ShapeIndex&) { in AddOutfeedToBuilder()
|
| D | xla_compiler.cc | 238 &out, [](Shape* subshape, const ShapeIndex&) { in BuildXlaCompilerSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 507 const ShapeIndex&) { in DummyCanShareBufferFunction()
1008 absl::flat_hash_map<ShapeIndex, GpuExecutable::OutputInfo>* output_info, in GetMlirAllocationInfo()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 528 [&](const xla::ShapeIndex&, XRTBufferAllocationPtr* buffer) { in MakeTuple()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 136 absl::Span<const std::pair<ShapeIndex, std::pair<int64, ShapeIndex>>> in CustomCallInternal()
|