Home
last modified time | relevance | path

Searched refs:IndexedArrayAnalysis (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc32 using Analysis = IndexedArrayAnalysis;
40 string IndexedArrayAnalysis::ToString(Array* root, bool print_constants) { in ToString()
79 StatusOr<Analysis::Array*> IndexedArrayAnalysis::GetArrayFor( in GetArrayFor()
90 Status IndexedArrayAnalysis::TraverseAndPopulateCache( in TraverseAndPopulateCache()
135 StatusOr<Analysis::Array*> IndexedArrayAnalysis::ComputeArrayFor( in ComputeArrayFor()
182 StatusOr<Analysis::Array*> IndexedArrayAnalysis::ComputeArrayForConstant( in ComputeArrayForConstant()
187 StatusOr<ScalarIndexedArray*> IndexedArrayAnalysis::FoldGatherOfGather( in FoldGatherOfGather()
253 StatusOr<Analysis::Array*> IndexedArrayAnalysis::ComputeArrayForGather( in ComputeArrayForGather()
466 IndexedArrayAnalysis::ReshapeToRemoveDegenerateDims( in ReshapeToRemoveDegenerateDims()
522 StatusOr<ScalarIndexedArray*> IndexedArrayAnalysis::ReshapeToAddDegenerateDims( in ReshapeToAddDegenerateDims()
[all …]
Dindexed_array_analysis.h33 class IndexedArrayAnalysis {
86 friend class IndexedArrayAnalysis; variable
102 friend class IndexedArrayAnalysis; variable
123 friend class IndexedArrayAnalysis; variable
208 friend class IndexedArrayAnalysis; variable
233 friend class IndexedArrayAnalysis; variable
Dindexed_array_analysis_test.cc62 IndexedArrayAnalysis indexed_tensor_analysis; in AssertArrayForRootExpressionIsImpl()
66 TF_ASSERT_OK_AND_ASSIGN(IndexedArrayAnalysis::Array* const array_result, in AssertArrayForRootExpressionIsImpl()