Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover.cc51 instruction->opcode() == HloOpcode::kTranspose; in IsReshapeOrTranspose()
124 case HloOpcode::kTranspose: in AreEquivalentReshapes()
153 CHECK(first_reshape_operand->opcode() == HloOpcode::kTranspose); in UpdateOperand()
168 case HloOpcode::kTranspose: { in UpdateOperand()
247 case HloOpcode::kTranspose: in PerformSinkReshapeOrTranspose()
Dtranspose_folding.cc51 if (transpose.opcode() != HloOpcode::kTranspose) { in CanFoldOperandsIntoDot()
96 if (operand.opcode() == HloOpcode::kTranspose) { in CanFoldOperandsIntoConvolution()
124 CHECK_EQ(new_lhs->opcode(), HloOpcode::kTranspose); in FoldTransposeIntoDot()
131 CHECK_EQ(new_rhs->opcode(), HloOpcode::kTranspose); in FoldTransposeIntoDot()
Dwhile_loop_all_reduce_code_motion.cc109 case HloOpcode::kTranspose: in IsAllReduceMovable()
157 case HloOpcode::kTranspose: in IsAllReduceMovable()
241 case HloOpcode::kTranspose: in IsAllReduceMovable()
291 case HloOpcode::kTranspose: in IsAllReduceMovable()
Dbfloat16_support.cc99 case HloOpcode::kTranspose: in EffectiveOperandPrecisionIsOutputPrecision()
Dfusion_node_indexing_evaluation.cc165 if (instruction->opcode() == HloOpcode::kTranspose || in UpdateIndexingUsersOfOperands()
Dhlo_opcode.h150 V(kTranspose, "transpose", 1) \
Dhlo_cse.cc141 case HloOpcode::kTranspose: in CseHash()
/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.cc124 {{OperatorType::kTranspose, 1}, "1.6.0"}, in GetMinimumRuntimeVersionForModel()
125 {{OperatorType::kTranspose, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
126 {{OperatorType::kTranspose, 3}, "1.15.0"}, in GetMinimumRuntimeVersionForModel()
127 {{OperatorType::kTranspose, 5}, kPendingReleaseOpVersion}, in GetMinimumRuntimeVersionForModel()
/external/tensorflow/tensorflow/core/kernels/rnn/
Dblas_gemm.cc46 se::blas::Transpose::kTranspose}; in operator ()()
/external/tensorflow/tensorflow/stream_executor/
Dblas.cc27 case Transpose::kTranspose: in TransposeString()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc121 auto lhs_transpose = lhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
123 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
Dtriangular_solve_thunk.cc60 return se::blas::Transpose::kTranspose; in TriangularSolveThunk()
/external/skia/src/sksl/codegen/
DSkSLGLSLCodeGenerator.h223 kTranspose enumerator
/external/skqp/src/sksl/
DSkSLGLSLCodeGenerator.h239 kTranspose enumerator
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_impl.h328 se::blas::Transpose::kTranspose,
420 auto gemv_trans_a = blas_transpose_a == se::blas::Transpose::kTranspose
422 : se::blas::Transpose::kTranspose;
500 se::blas::Transpose::kTranspose,

123