Searched refs:indexed_array (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 66 auto* indexed_array = root->as<ScalarIndexedArray>(); in ToString() local
71 "(", name, " ", ToString(indexed_array->source(), print_constants), in ToString()
72 " ", ToString(indexed_array->indices(), print_constants), " ", in ToString()
73 indexed_array->source_dim(), "->[", in ToString()
74 StrJoin(indexed_array->output_dims(), ","), "])"); in ToString()
1000 absl::string_view tag, Analysis::ScalarIndexedConstantArray* indexed_array, in CanFoldDotIntoIndexedArray() argument
1004 GetOnlyNonContractingNonBatchDim(indexed_array->shape().rank(), in CanFoldDotIntoIndexedArray()
1011 if (indexed_array->output_dims().size() != 1 || in CanFoldDotIntoIndexedArray()
1012 indexed_array->output_dims()[0] != *non_contracting_non_batch_dim) { in CanFoldDotIntoIndexedArray()
1017 int64 indexed_array_rank = indexed_array->shape().rank(); in CanFoldDotIntoIndexedArray()
[all …]
|