Searched refs:ScalarIndexedArray (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.h | 175 class ScalarIndexedArray : public Array {
194 explicit ScalarIndexedArray(Array* source, Array* indices, int64 source_dim, in ScalarIndexedArray() function
215 class ScalarIndexedConstantArray : public ScalarIndexedArray {
228 : ScalarIndexedArray(source, indices, source_dim, in ScalarIndexedConstantArray()
304 StatusOr<ScalarIndexedArray*> FoldGatherOfGather(
305 ScalarIndexedArray* source, Array* indices, int64 source_dim,
310 StatusOr<ScalarIndexedArray*> ReshapeToRemoveDegenerateDims(
311 ScalarIndexedArray* operand);
315 StatusOr<ScalarIndexedArray*> ReshapeToAddDegenerateDims(
316 ScalarIndexedArray* operand, absl::Span<const int64> degenerate_dims);
[all …]
|
| D | indexed_array_analysis.cc | 36 using ScalarIndexedArray = Analysis::ScalarIndexedArray; typedef
66 auto* indexed_array = root->as<ScalarIndexedArray>(); in ToString()
187 StatusOr<ScalarIndexedArray*> IndexedArrayAnalysis::FoldGatherOfGather( in FoldGatherOfGather()
188 ScalarIndexedArray* source, Array* indices, int64 source_dim, in FoldGatherOfGather()
298 if (auto* indexed = dynamic_cast<ScalarIndexedArray*>(source)) { in ComputeArrayForGather()
308 return Construct<ScalarIndexedArray>(source, indices, source_dim, output_dims, in ComputeArrayForGather()
465 StatusOr<ScalarIndexedArray*>
467 ScalarIndexedArray* operand) { in ReshapeToRemoveDegenerateDims()
522 StatusOr<ScalarIndexedArray*> IndexedArrayAnalysis::ReshapeToAddDegenerateDims( in ReshapeToAddDegenerateDims()
523 ScalarIndexedArray* operand, absl::Span<const int64> degenerate_dims) { in ReshapeToAddDegenerateDims()
[all …]
|