| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_computation.h | 265 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index,
440 HloInstruction* instruction, ShapeIndex* index,
442 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index,
|
| copy_insertion.cc | 148 const ShapeIndex& index = pair.first;
176 const ShapeIndex& index = pair.first;
201 // Elements (each ShapeIndex) in the loop state are considered independently. A
298 ShapeIndex current_index;
353 [&](const ShapeIndex& output_index,
400 [&](const ShapeIndex& output_index,
[all...] |
| buffer_assignment.cc | 66 const HloInstruction* instruction, const ShapeIndex& index,
291 StrAppend(&output, ", parameter ", parameter_number(), " at ShapeIndex ",
358 const HloInstruction* instruction, const ShapeIndex& index) const {
376 const HloInstruction* hlo, const ShapeIndex& shape_index) const {
395 const ShapeIndex& index) const {
411 const HloInstruction* instruction, const ShapeIndex& index) const {
448 const HloInstruction* hlo_a, const ShapeIndex& shape_index_a,
449 const HloInstruction* hlo_b, const ShapeIndex& shape_index_b) const {
463 [&](const Shape& /*subshape*/, const ShapeIndex& index) {
[all...] |
| while_loop_invariant_code_motion.cc | 223 [&input_size](const Shape& subshape, const ShapeIndex& /*index*/) {
231 [&output_size](const Shape& subshape, const ShapeIndex& /*index*/) {
|
| hlo_alias_analysis.cc | 191 ShapeIndex output_index = pos.index;
347 const HloInstruction* instruction, const ShapeIndex& index) const {
354 const HloInstruction* instruction, const ShapeIndex& index) {
361 const HloInstruction* instruction, const ShapeIndex& index) const {
461 [&out, &instruction, this](const Shape&, const ShapeIndex& index) {
|
| transfer_manager.cc | 245 [&](const Shape& device_subshape, const ShapeIndex& index) -> Status {
252 ShapeIndex element_index = index;
326 const ShapeIndex& index = pair.first;
|
| hlo_sharding.h | 174 HloSharding GetSubSharding(const Shape& shape, const ShapeIndex& index) const;
|
| shaped_buffer.cc | 86 [this, &s](const Shape& subshape, const ShapeIndex& index) {
|
| hlo_sharding.cc | 54 flattened_list.push_back(sub_shardings.element(ShapeIndex({})));
228 *result.mutable_element(ShapeIndex({})) = *it;
447 const ShapeIndex& index) const {
|
| hlo_computation.cc | 625 HloInstruction* instruction, ShapeIndex* index,
627 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index,
674 ShapeIndex index;
676 HloInstruction* leaf, const ShapeIndex& leaf_index,
696 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index,
703 ShapeIndex index;
[all...] |
| instruction_fusion.cc | 176 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) {
275 [&size](const Shape& subshape, const ShapeIndex& shape_index) {
|
| layout_assignment.cc | 110 [&](const ShapeIndex&, const PointsToSet::BufferList& buffers) {
135 [&buffer_set](const ShapeIndex& /*index*/,
305 const ShapeIndex& index) -> Status {
[all...] |
| layout_assignment.h | 458 const ShapeIndex& index);
|
| heap_simulator_test.cc | 322 const ShapeIndex& index) const {
327 int64 OffsetAt(const HloInstruction* instruction, const ShapeIndex& index) {
340 const ShapeIndex& index_a,
342 const ShapeIndex& index_b) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| infeed_thunk.cc | 57 ShapeIndex index = {0};
|
| ir_emitter.h | 124 const ShapeIndex& shape_index = {}) {
139 const HloInstruction& hlo, const ShapeIndex& index = {}) const {
|
| gpu_layout_assignment.cc | 349 buf.index() == ShapeIndex({0})) {
355 buf.index() == ShapeIndex({0})) {
|
| hlo_to_ir_bindings.cc | 98 [&](const Shape& /*subshape*/, const ShapeIndex& index) {
215 const ShapeIndex& shape_index) {
|
| gpu_executable.cc | 298 const ShapeIndex& index, se::DeviceMemoryBase* device_memory) {
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| xrt_execute_op.cc | 239 [&](const xla::ShapeIndex& output_index,
259 xla::ShapeIndex shape_index;
|
| xrt_state_ops.h | 138 [&](const xla::ShapeIndex& index,
337 xla::ShapeIndex shape_index;
553 [&](xla::Shape* subshape, const xla::ShapeIndex& index) -> Status {
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal_util.cc | 51 &result_shape, [](Shape* subshape, const ShapeIndex&) {
64 [&](const Shape& subshape, const ShapeIndex& shape_index) {
|
| shape_util_test.cc | 35 ShapeIndex index = {1, 2, 3, 4};
479 shape, [&calls, &shape](const Shape& subshape, const ShapeIndex& index) {
494 shape, [&calls, &shape](const Shape& subshape, const ShapeIndex& index) {
516 &shape, [&calls, &shape](const Shape* subshape, const ShapeIndex& index) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_executable.cc | 223 [&](const ShapeIndex& index, se::DeviceMemoryBase* device_memory) {
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_library_test_base.cc | 332 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) {
389 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) {
624 [](Shape* subshape, const ShapeIndex&) {
|
