tf_1.8_xla_doc
Namespaces | Functions
transpose_folding.cc File Reference
#include "tensorflow/compiler/xla/service/transpose_folding.h"
#include <vector>
#include "tensorflow/compiler/xla/service/dfs_hlo_visitor_with_default.h"
#include "tensorflow/compiler/xla/service/hlo_computation.h"
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
#include "tensorflow/compiler/xla/shape_util.h"
#include "tensorflow/compiler/xla/status_macros.h"
#include "tensorflow/compiler/xla/util.h"
#include "tensorflow/core/lib/core/errors.h"
#include "tensorflow/core/lib/core/status.h"
#include "tensorflow/core/platform/logging.h"
Include dependency graph for transpose_folding.cc:

Namespaces

 xla
 namespace for xla
 

Functions

TransposeFolding::OperandIndices xla::anonymous_namespace{transpose_folding.cc}::CanFoldOperandsIntoDot (const HloInstruction &dot, const TransposeFolding::TransposableGemmOperandsFn &transposable_gemm_operands)
 
bool xla::anonymous_namespace{transpose_folding.cc}::FoldTransposeIntoDot (InstructionOperandsPair pair)