| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover.cc | 53 instruction->opcode() == HloOpcode::kTranspose; in IsReshapeOrTranspose()
120 case HloOpcode::kTranspose: in AreEquivalentReshapes()
206 CHECK(first_reshape_operand->opcode() == HloOpcode::kTranspose); in UpdateOperand()
221 case HloOpcode::kTranspose: { in UpdateOperand()
309 case HloOpcode::kTranspose: in TrySinkReshapeOrTranspose()
|
| D | bfloat16_support.cc | 91 case HloOpcode::kTranspose: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | hlo_opcode.h | 128 V(kTranspose, "transpose") \
|
| D | transpose_folding.cc | 64 if (operand.opcode() == HloOpcode::kTranspose) { in CanFoldOperandsIntoConvolution()
|
| D | while_loop_invariant_code_motion.cc | 114 case HloOpcode::kTranspose: in NotWorthHoistingIndividually()
|
| D | hlo_tfgraph_builder.cc | 180 case HloOpcode::kTranspose: in SetNodeAttrs()
|
| D | hlo_instruction.cc | 760 WrapUnique(new HloInstruction(HloOpcode::kTranspose, shape)); in CreateTranspose()
1360 case HloOpcode::kTranspose: in CloneWithNewOperands()
1587 opcode() == HloOpcode::kTranspose); in CanHaveDimensionsField()
1810 case HloOpcode::kTranspose: in IdenticalSlowPath()
1861 return (opcode_ == HloOpcode::kTranspose) && in IsRank2Transpose()
2393 if (opcode() == HloOpcode::kTranspose || opcode() == HloOpcode::kCopy || in ToCategory()
2636 case HloOpcode::kTranspose: in Visit()
3148 case HloOpcode::kTranspose: in OperandElementUse()
3380 case HloOpcode::kTranspose: in CouldBeBitcast()
|
| D | instruction_fusion.cc | 79 case HloOpcode::kTranspose: in IsExpensive()
|
| D | algebraic_simplifier.cc | 68 CHECK_EQ(HloOpcode::kTranspose, transpose->opcode()); in TransposeIsBitcast()
1087 case HloOpcode::kTranspose: in OutputIsPermutationOfOperandElements()
1626 arg->opcode() == HloOpcode::kTranspose) { in HandleReduce()
1835 if (HloOpcode::kTranspose == operand->opcode()) { in HandleTranspose()
|
| D | layout_assignment.cc | 950 if (instruction->opcode() == HloOpcode::kTranspose) { in ChooseOperandLayoutFromOutputLayout()
1022 if (user->opcode() == HloOpcode::kTranspose) { in ChooseOutputLayoutFromOperandLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization_test.cc | 96 EXPECT_EQ(HloOpcode::kTranspose, output_reshape->opcode()); in TEST_F()
100 EXPECT_EQ(HloOpcode::kTranspose, input_reshape->opcode()); in TEST_F()
102 EXPECT_EQ(HloOpcode::kTranspose, kernel_reshape->opcode()); in TEST_F()
|
| D | cpu_instruction_fusion.cc | 42 hlo.opcode() == HloOpcode::kTranspose; in CanBeLoopFused()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 59 auto lhs_transpose = lhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemm()
61 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemm()
102 auto lhs_transpose = lhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
104 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
|
| D | instruction_fusion.cc | 40 hlo.opcode() == HloOpcode::kTranspose; in IsFusile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_matmul_op_impl.h | 292 constexpr perftools::gputools::blas::Transpose kTranspose =
295 : perftools::gputools::blas::Transpose::kTranspose;
297 perftools::gputools::blas::Transpose::kNoTranspose, kTranspose};
353 blas_transpose_a == perftools::gputools::blas::Transpose::kTranspose
355 : perftools::gputools::blas::Transpose::kTranspose;
|
| D | conv_grad_ops_3d.cc | 487 auto transpose = perftools::gputools::blas::Transpose::kTranspose; in Compute()
515 auto transpose = perftools::gputools::blas::Transpose::kTranspose; in Compute()
848 perftools::gputools::blas::Transpose::kTranspose, in Compute()
874 perftools::gputools::blas::Transpose::kTranspose, in Compute()
|
| D | matmul_op.cc | 170 trans ? perftools::gputools::blas::Transpose::kTranspose in Compute()
270 Transpose trans[] = {Transpose::kNoTranspose, Transpose::kTranspose}; in launch()
|
| D | conv_grad_filter_ops.cc | 722 perftools::gputools::blas::Transpose::kTranspose, n, in operator ()()
752 perftools::gputools::blas::Transpose::kTranspose, n, in operator ()()
|
| D | conv_grad_input_ops.cc | 778 auto transpose = perftools::gputools::blas::Transpose::kTranspose; in operator ()()
810 auto transpose = perftools::gputools::blas::Transpose::kTranspose; in operator ()()
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| D | resolve_transpose_attributes.cc | 29 if (op_it->get()->type != OperatorType::kTranspose) return false; in Run()
|
| D | convert_trivial_transpose_to_reshape.cc | 26 if (transpose_it->get()->type != OperatorType::kTranspose) { in Run()
|
| D | resolve_constant_transpose.cc | 107 if (base_op->type != OperatorType::kTranspose) { in Run()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.cc | 28 case Transpose::kTranspose: in TransposeString()
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | blas_gemm.cc | 46 perftools::gputools::blas::Transpose::kTranspose}; in operator ()()
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| D | model.h | 116 kTranspose, enumerator
1027 TransposeOperator() : Operator(OperatorType::kTranspose) {}
|