| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter_test.cc | 266 HloInstruction::CreateReverse(kernel->shape(), kernel, {2, 3})); in TEST_F()
415 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
466 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
521 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
578 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
|
| D | cudnn_conv_rewriter.cc | 445 reverse_filter = c->AddInstruction(HloInstruction::CreateReverse( in MatchBackwardInput()
448 reverse_filter = c->AddInstruction(HloInstruction::CreateReverse( in MatchBackwardInput()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 471 auto reverse1 = builder.AddInstruction(HloInstruction::CreateReverse( in XLA_TEST_F()
487 auto reverse1 = builder.AddInstruction(HloInstruction::CreateReverse( in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 623 HloInstruction::CreateReverse(data_shape_, gte11, {0})); in BuildNestedBodyComputation()
1738 HloInstruction::CreateReverse(element_shape, body_element_3, {0})); in TEST_F()
1891 HloInstruction::CreateReverse(element_shape, element_1, {0})); in MakeBenchmarkWhileBody()
|
| D | tuple_points_to_analysis_test.cc | 1163 HloInstruction::CreateReverse(data_shape, operand, {0, 1})); in TEST_F()
1248 builder.AddInstruction(HloInstruction::CreateReverse(shape, param, {0})); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 1430 HloInstruction::CreateReverse(vector_shape_, negate, {0})); in TEST_P()
2105 HloInstruction::CreateReverse(data_shape, neg, {0, 1})); in TEST_F()
2542 HloInstruction::CreateReverse(data_shape, operand, {0, 1})); in TEST_F()
2654 builder.AddInstruction(HloInstruction::CreateReverse(shape, param, {0})); in TEST_F()
|
| D | buffer_liveness_test.cc | 214 builder.AddInstruction(HloInstruction::CreateReverse(vec_, negate, {0})); in TEST_F()
|
| D | hlo_instruction.h | 786 static std::unique_ptr<HloInstruction> CreateReverse(
|
| D | buffer_assignment_test.cc | 1700 HloInstruction::CreateReverse(f32vec100_, log, {0})); in TEST_F()
1767 builder.AddInstruction(HloInstruction::CreateReverse( in TEST_F()
|
| D | hlo_evaluator_test.cc | 1035 rhs_instruction = b.AddInstruction(HloInstruction::CreateReverse( in TEST_P()
2012 b.AddInstruction(HloInstruction::CreateReverse(shape, operand, {0, 1})); in TEST_P()
|
| D | hlo_instruction.cc | 203 instruction = CreateReverse(shape, operands(0), in CreateFromProto()
908 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReverse( in CreateReverse() function in xla::HloInstruction
|
| D | hlo_parser.cc | 1228 HloInstruction::CreateReverse(shape, operands[0], *dimensions)); in ParseInstructionRhs()
|
| D | algebraic_simplifier_test.cc | 3743 HloInstruction::CreateReverse(shape, a, /*dimensions=*/{2, 3})); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 449 HloInstruction::CreateReverse(param_shape, param0, {0})); in TEST_F()
|