Home
last modified time | relevance | path

Searched refs:kTranspose (Results 1 – 25 of 56) 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()
Ddepthwise_convolution_converter_test.cc85 EXPECT_EQ(transpose->opcode(), HloOpcode::kTranspose) in TEST_F()
Dbfloat16_support.cc98 case HloOpcode::kTranspose: in EffectiveOperandPrecisionIsOutputPrecision()
Dhlo_opcode.h143 V(kTranspose, "transpose", 1) \
Dar_crs_combiner.cc49 case HloOpcode::kTranspose: in MatchesArCrsPattern()
457 case HloOpcode::kTranspose: in RewriteGraph()
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.cc42 hlo.opcode() == HloOpcode::kTranspose; in CanBeLoopFused()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dremove_successive_transpose.cc64 if (op->get()->type != OperatorType::kTranspose) { in Run()
73 if (!next || next->type != OperatorType::kTranspose) { in Run()
Dpropagate_fake_quant_num_bits.cc105 case OperatorType::kTranspose: in DoesOpBlockBackwardPropagation()
133 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.cc57 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/lite/toco/tflite/
Dop_version.cc116 {{OperatorType::kTranspose, 1}, "1.6.0"}, in GetMinimumRuntimeVersionForModel()
117 {{OperatorType::kTranspose, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
118 {{OperatorType::kTranspose, 3}, "1.15.0"}, in GetMinimumRuntimeVersionForModel()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h302 constexpr se::blas::Transpose kTranspose =
304 : se::blas::Transpose::kTranspose;
306 kTranspose};
379 auto gemv_trans_a = blas_transpose_a == se::blas::Transpose::kTranspose
381 : se::blas::Transpose::kTranspose;
437 constexpr perftools::gputools::blas::Transpose kTranspose =
440 : perftools::gputools::blas::Transpose::kTranspose;
442 perftools::gputools::blas::Transpose::kNoTranspose, kTranspose};
/external/tensorflow/tensorflow/stream_executor/
Dblas.cc27 case Transpose::kTranspose: in TransposeString()
/external/tensorflow/tensorflow/core/kernels/rnn/
Dblas_gemm.cc46 se::blas::Transpose::kTranspose}; in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc108 auto lhs_transpose = lhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
110 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
Dtriangular_solve_thunk.cc60 return se::blas::Transpose::kTranspose; in TriangularSolveThunk()
/external/skqp/src/sksl/
DSkSLGLSLCodeGenerator.h239 kTranspose enumerator
/external/skia/src/sksl/
DSkSLGLSLCodeGenerator.h257 kTranspose enumerator

123