| /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
177 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 | 235 const ShapeIndex& /*index*/) { in TryHoistingInvariantInstructionsFromWhileBody()
244 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 | optimize_input_output_buffer_alias_test.cc | 48 [&](const ShapeIndex&, const HloInputOutputAliasConfig::Alias&) { in AliasCount()
|
| D | dynamic_dimension_inference.cc | 1170 hlo, [&](HloInstruction*, ShapeIndex /*index*/, int64_t dimension, in HandleDynamicSlice() argument
1194 [&](HloInstruction* /*operand*/, ShapeIndex /*index*/, int64_t dimension, in HandleDynamicUpdateSlice() argument
1195 int64_t operand_index, HloInstruction* dynamic_size) { in HandleDynamicUpdateSlice()
1230 HloInstruction* dynamic_size) { in HandleGather()
1320 HloInstruction* dynamic_size) -> Status { in HandleConditional()
1487 int64_t operand_index, HloInstruction* operand_dynamic_size) { in HandleScatter()
|
| D | while_loop_expensive_invariant_code_motion.cc | 283 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 | 853 subshape, [&](const Shape& s, const ShapeIndex& /*index*/) { in UnflattenTupleInstr()
915 [&](const Shape& s, const ShapeIndex& /*index*/) { in TryFlattenNestedTuples()
|
| D | hlo_memory_scheduler.cc | 124 const PointsToSet::BufferList& buffers) { in ListScheduler()
|
| D | layout_assignment.cc | 118 [&](const ShapeIndex&, const PointsToSet::BufferList& buffers) { in LayoutConstraints()
144 const PointsToSet::BufferList& buffers) { in GetBufferSet()
|
| /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 | 109 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()
622 [](Shape* subshape, const ShapeIndex&) { in MaybeConvertShapeToBfloat16()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | run_hlo_module.cc | 83 const ShapeIndex& /*shape_index*/) { in OnMiscompare()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 181 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
|
| D | xla.cc | 147 &shape_with_layout, [](Shape* subshape, const ShapeIndex&) { in PYBIND11_MODULE()
|
| D | outfeed_receiver.cc | 431 &shape_with_layout, [](Shape* subshape, const ShapeIndex&) { in AddOutfeedToBuilder()
|
| D | xla_client.py | 288 ShapeIndex = _xla.ShapeIndex variable
|
| D | xla_compiler.cc | 288 &out, [](Shape* subshape, const ShapeIndex&) { in BuildXlaCompilerSubmodule()
|
| /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 | gpu_compiler.cc | 648 const ShapeIndex&) { in DummyCanShareBufferFunction()
|