Home
last modified time | relevance | path

Searched refs:kTranspose (Results 1 – 25 of 52) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover.cc50 instruction->opcode() == HloOpcode::kTranspose; in IsReshapeOrTranspose()
123 case HloOpcode::kTranspose: in AreEquivalentReshapes()
152 CHECK(first_reshape_operand->opcode() == HloOpcode::kTranspose); in UpdateOperand()
167 case HloOpcode::kTranspose: { in UpdateOperand()
246 case HloOpcode::kTranspose: in PerformSinkReshapeOrTranspose()
Dtranspose_folding.cc67 if (operand.opcode() == HloOpcode::kTranspose) { in CanFoldOperandsIntoConvolution()
97 CHECK_EQ(new_lhs->opcode(), HloOpcode::kTranspose); in FoldTransposeIntoDot()
103 CHECK_EQ(new_rhs->opcode(), HloOpcode::kTranspose); in FoldTransposeIntoDot()
Dbfloat16_support.cc98 case HloOpcode::kTranspose: in EffectiveOperandPrecisionIsOutputPrecision()
Dar_crs_combiner.cc49 case HloOpcode::kTranspose: in MatchesArCrsPattern()
334 case HloOpcode::kTranspose: in RewriteGraph()
Ddynamic_padder.cc80 case HloOpcode::kTranspose: in ChooseIdentityValue()
Dhlo_opcode.h141 V(kTranspose, "transpose", 1) \
Dwhile_loop_invariant_code_motion.cc119 case HloOpcode::kTranspose: in NotWorthHoistingIndividually()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization_test.cc102 EXPECT_EQ(HloOpcode::kTranspose, output_reshape->opcode()); in TEST_F()
106 EXPECT_EQ(HloOpcode::kTranspose, input_reshape->opcode()); in TEST_F()
108 EXPECT_EQ(HloOpcode::kTranspose, kernel_reshape->opcode()); in TEST_F()
Dcpu_instruction_fusion.cc43 hlo.opcode() == HloOpcode::kTranspose; in CanBeLoopFused()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h294 constexpr se::blas::Transpose kTranspose =
296 : se::blas::Transpose::kTranspose;
298 kTranspose};
353 auto gemv_trans_a = blas_transpose_a == se::blas::Transpose::kTranspose
355 : se::blas::Transpose::kTranspose;
410 constexpr perftools::gputools::blas::Transpose kTranspose =
413 : perftools::gputools::blas::Transpose::kTranspose;
415 perftools::gputools::blas::Transpose::kNoTranspose, kTranspose};
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc66 auto lhs_transpose = lhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemm()
68 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemm()
131 auto lhs_transpose = lhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
133 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
Dtriangular_solve_thunk.cc60 return se::blas::Transpose::kTranspose; in TriangularSolveThunk()
Dgpu_fusible.cc180 instr.opcode() == HloOpcode::kTranspose; in IsLoopFusible()
Dinstruction_fusion.cc179 if (producer->opcode() == HloOpcode::kTranspose) { in ShouldFuseInexpensiveChecks()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fake_quant_num_bits.cc109 case OperatorType::kTranspose: in DoesOpBlockBackwardPropagation()
137 case OperatorType::kTranspose: in DoesOpInputBlockBackwardPropagation()
Dresolve_transpose_attributes.cc32 if (op_it->get()->type != OperatorType::kTranspose) in Run()
Dconvert_trivial_transpose_to_reshape.cc56 if (transpose_it->get()->type != OperatorType::kTranspose) { in Run()
Dreorder_elementwise_unary.cc56 case OperatorType::kTranspose: in IsMoveOperator()
Dmerge_reshape_into_preceding_transpose.cc136 transpose_it->get(), OperatorType::kTranspose); in Run()
Dresolve_tensorflow_matmul.cc33 if (op->type != OperatorType::kTranspose) { in FindTransposeOpWithInput()
Dresolve_constant_transpose.cc110 if (base_op->type != OperatorType::kTranspose) { in Run()
/external/tensorflow/tensorflow/stream_executor/
Dblas.cc27 case Transpose::kTranspose: in TransposeString()
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dblas_gemm.cc46 se::blas::Transpose::kTranspose}; in operator ()()
/external/skqp/src/sksl/
DSkSLGLSLCodeGenerator.h239 kTranspose enumerator
/external/skia/src/sksl/
DSkSLGLSLCodeGenerator.h241 kTranspose enumerator

123