HomeSort by relevance Sort by last modified time
    Searched refs:IndexedArrayAnalysis (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
indexed_array_analysis.h 29 // IndexedArrayAnalysis decides if an HLO instruction can be rewritten as a
31 // instances of IndexedArrayAnalysis::Array, which can be inspected to discover
33 class IndexedArrayAnalysis {
35 // IndexedArrayAnalysis maps each HLO instruction to an instance of a Array.
73 // IndexedArrayAnalysis. Instances of UnknownArray just wrap an existing
86 friend class IndexedArrayAnalysis;
102 friend class IndexedArrayAnalysis;
123 friend class IndexedArrayAnalysis;
208 friend class IndexedArrayAnalysis;
233 friend class IndexedArrayAnalysis;
    [all...]
indexed_array_analysis.cc 32 using Analysis = IndexedArrayAnalysis;
40 string IndexedArrayAnalysis::ToString(Array* root, bool print_constants) {
79 StatusOr<Analysis::Array*> IndexedArrayAnalysis::GetArrayFor(
90 Status IndexedArrayAnalysis::TraverseAndPopulateCache(
135 StatusOr<Analysis::Array*> IndexedArrayAnalysis::ComputeArrayFor(
182 StatusOr<Analysis::Array*> IndexedArrayAnalysis::ComputeArrayForConstant(
187 StatusOr<ScalarIndexedArray*> IndexedArrayAnalysis::FoldGatherOfGather(
253 StatusOr<Analysis::Array*> IndexedArrayAnalysis::ComputeArrayForGather(
466 IndexedArrayAnalysis::ReshapeToRemoveDegenerateDims(
522 StatusOr<ScalarIndexedArray*> IndexedArrayAnalysis::ReshapeToAddDegenerateDims
    [all...]
indexed_array_analysis_test.cc 62 IndexedArrayAnalysis indexed_tensor_analysis;
66 TF_ASSERT_OK_AND_ASSIGN(IndexedArrayAnalysis::Array* const array_result,

Completed in 609 milliseconds