Searched defs:source_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.h | 186 int64 source_dim() const { return source_dim_; } in source_dim() function
194 explicit ScalarIndexedArray(Array* source, Array* indices, int64 source_dim, in ScalarIndexedArray()
225 int64 source_dim, in ScalarIndexedConstantArray()
337 Array* source, Array* indices, int64 source_dim, in ConstructScalarIndexedArray()
|
| D | indexed_array_analysis.cc | 188 ScalarIndexedArray* source, Array* indices, int64 source_dim, in FoldGatherOfGather()
290 int64 source_dim = dim_numbers.start_index_map(0); in ComputeArrayForGather() local
|