Home
last modified time | relevance | path

Searched refs:CreateTranspose (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc78 HloInstruction::CreateTranspose(new_input_shape, input, in Run()
100 HloInstruction::CreateTranspose(new_kernel_shape, kernel, in Run()
143 hlo, HloInstruction::CreateTranspose( in Run()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dtranspose_test_util.cc51 GPUOperation operation = CreateTranspose(op_def, attr); in TransposeIntTest()
93 GPUOperation operation = CreateTranspose(op_def, attr); in TransposeUintTest()
132 GPUOperation operation = CreateTranspose(op_def, attr); in TransposeTest()
Dtranspose.h26 GPUOperation CreateTranspose(const OperationDef& definition,
Dtranspose.cc95 GPUOperation CreateTranspose(const OperationDef& definition, in CreateTranspose() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc118 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
179 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
281 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
330 HloInstruction* lhs = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
641 HloInstruction::CreateTranspose(shape, body_param, {0, 1})); in TEST_F()
646 builder_f.AddInstruction(HloInstruction::CreateTranspose(shape, a_f, {0, 1})); in TEST_F()
724 HloInstruction::CreateTranspose(shape, body_dot2, {0, 1})); in TEST_F()
931 builder.AddInstruction(HloInstruction::CreateTranspose(shape, a, {0, 1})); in TEST_F()
933 builder.AddInstruction(HloInstruction::CreateTranspose(shape, b, {0, 1})); in TEST_F()
984 HloInstruction::CreateTranspose(shape, a_gte, {0, 1})); in TEST_F()
[all …]
Dtopk_rewriter.cc220 input = comp->AddInstruction(HloInstruction::CreateTranspose( in TransformToCustomCall()
243 value_gte = comp->AddInstruction(HloInstruction::CreateTranspose( in TransformToCustomCall()
246 index_gte = comp->AddInstruction(HloInstruction::CreateTranspose( in TransformToCustomCall()
Dtranspose_folding_test.cc235 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
293 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
357 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
426 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
Ddot_decomposer.cc85 HloInstruction::CreateTranspose( in CanonicalizeDot()
135 HloInstruction::CreateTranspose( in CanonicalizeDot()
Dreshape_mover.cc157 return computation->AddInstruction(HloInstruction::CreateTranspose( in UpdateOperand()
249 HloInstruction::CreateTranspose(instruction->shape(), new_elementwise, in PerformSinkReshapeOrTranspose()
Dlayout_assignment_test.cc365 HloInstruction::CreateTranspose(bshape, log, {1, 0})); in TEST_F()
403 HloInstruction::CreateTranspose(cshape, broadcast, {2, 1, 0})); in TEST_F()
446 HloInstruction::CreateTranspose(f32_43, broadcast, {1, 0})); in TEST_F()
549 auto transpose = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
567 auto transpose = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
587 HloInstruction::CreateTranspose(input_shape, param, {0, 2, 1})); in TEST_F()
1361 b.AddInstruction(HloInstruction::CreateTranspose(ashape, param0, {1, 0})); in TEST_F()
Dbitcast_decomposer.cc117 HloInstruction::CreateTranspose(shape, input, perm)); in Run()
Dlayout_normalization.cc313 HloInstruction::CreateTranspose(normalized_shape, a0, dimensions)); in HandleTranspose()
349 HloInstruction::CreateTranspose(s_normalized, a0, dimensions)); in HandleCopy()
Dhlo_computation_test.cc531 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
567 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
604 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
Dhlo_instruction_test.cc1308 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
1407 builder.AddInstruction(HloInstruction::CreateTranspose(s, y, {1, 0})); in TEST_F()
1458 HloInstruction::CreateTranspose(data_shape, b, {1, 0})); in TEST_F()
1539 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
1786 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
1833 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
1889 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
Dhlo_constant_folding_test.cc191 HloInstruction::CreateTranspose(shape, literal_instruction, permutation)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc90 HloInstruction::CreateTranspose(param_shape, lhs, {1, 0})); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc206 auto reshape3 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F()
438 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F()
453 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F()
578 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc101 HloInstruction::CreateTranspose(ShapeUtil::MakeShape(F32, {}), exp1, {})); in TEST_F()
137 auto transpose2 = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
Dgpu_conv_rewriter.cc542 rhs = c->AddInstruction(HloInstruction::CreateTranspose( in MatchBackwardInput()
639 lhs = add(HloInstruction::CreateTranspose( in ConvertBatchGroupedToFeatureGroupedConvolution()
/external/armnn/src/backends/backendsCommon/
DWorkloadFactoryBase.hpp271 std::unique_ptr<IWorkload> CreateTranspose(const TransposeQueueDescriptor& /*descriptor*/, in CreateTranspose() function in armnn::WorkloadFactoryBase
/external/armnn/include/armnn/backends/
DWorkloadFactory.hpp400 virtual std::unique_ptr<IWorkload> CreateTranspose(const TransposeQueueDescriptor& descriptor,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.cc221 GPUOperation operation = CreateTranspose(op_def, attr); in SelectTranspose()
/external/armnn/src/backends/neon/
DNeonWorkloadFactory.hpp356 std::unique_ptr<IWorkload> CreateTranspose(const TransposeQueueDescriptor& descriptor,
/external/armnn/src/backends/reference/
DRefWorkloadFactory.hpp380 std::unique_ptr<IWorkload> CreateTranspose(const TransposeQueueDescriptor& descriptor,
/external/armnn/src/backends/cl/
DClWorkloadFactory.hpp359 std::unique_ptr<IWorkload> CreateTranspose(const TransposeQueueDescriptor& descriptor,

12