| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder.h | 295 ShapeIndex dynamic_size_param_index,
297 ShapeIndex target_param_index, int64 target_dim_num);
302 void SetUpAlias(const ShapeIndex& output_index, int64 param_number,
303 const ShapeIndex& param_index) {
310 ShapeIndex output_index;
314 ShapeIndex param_index;
[all...] |
| xla_builder_test.cc | [all...] |
| xla_builder.cc | 256 ShapeIndex dynamic_size_param_index,
258 ShapeIndex target_param_index,
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_instruction.h | [all...] |
| tuple_points_to_analysis_test.cc | 65 const ShapeIndex& index) {
106 const HloInstruction* instruction, const ShapeIndex& index,
107 absl::Span<const std::pair<HloInstruction*, ShapeIndex>> expected) {
[all...] |
| buffer_liveness_test.cc | 41 const ShapeIndex& index) {
67 const ShapeIndex& index) {
[all...] |
| hlo_cost_analysis.cc | 584 [&](const Shape& subshape, const ShapeIndex&) {
625 [this](const Shape& subshape, const ShapeIndex& /*shape_index*/) {
|
| while_loop_simplifier.cc | 603 subshape, [&](const Shape& s, const ShapeIndex& /*index*/) {
665 [&](const Shape& s, const ShapeIndex& /*index*/) {
[all...] |
| hlo_verifier.cc | 33 shape, [](const Shape& subshape, const ShapeIndex&) -> Status {
788 [&](const Shape& subshape, const ShapeIndex& index) {
[all...] |
| hlo_memory_scheduler.cc | 120 [&](const ShapeIndex& /*index*/,
|
| hlo_alias_analysis_test.cc | 59 const ShapeIndex& index = {}) const {
85 const ShapeIndex& index = {}) const {
[all...] |
| hlo_rematerialization.cc | 529 // proper ShapeIndex.
[all...] |
| hlo_graph_dumper.cc | [all...] |
| buffer_assignment_test.cc | 263 const ShapeIndex& index) {
[all...] |
| hlo_instruction.cc | [all...] |
| hlo_instructions.h | 666 const ShapeIndex& shape_index = {});
[all...] |
| hlo_dataflow_analysis_test.cc | 61 const ShapeIndex& index = {}) {
[all...] |
| hlo_instructions.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.h | 192 const HloInstruction& hlo, const ShapeIndex& index = {}) const {
|
| ir_emitter.cc | 531 ShapeIndex shape_index =
532 sort->values_count() > 0 ? ShapeIndex({i}) : ShapeIndex({});
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal_comparison.cc | 736 // via recursion. shape_index is the ShapeIndex of expected (or actual)
741 const ShapeIndex& shape_index) {
749 ShapeIndex element_index = shape_index;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 191 subshape, [&](const Shape& s, const ShapeIndex& index) {
192 ShapeIndex original(index);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| local_client_execute_test.cc | 537 ShapeIndex index;
[all...] |
