| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 269 HloInstruction::CreateReverse(kernel->shape(), kernel, {2, 3})); in TEST_F()
422 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
474 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
530 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
588 HloInstruction::CreateReverse(kernel->shape(), kernel, {0, 1})); in TEST_F()
|
| D | gpu_conv_rewriter.cc | 476 HloInstruction::CreateReverse(reverse_filter->shape(), reverse_filter, in MatchBackwardInput()
479 HloInstruction::CreateReverse(reverse_filter->shape(), reverse_filter, in MatchBackwardInput()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_normalization.cc | 372 HloInstruction::CreateReverse(a0->shape(), a0, new_dimensions)); in HandleReverse()
|
| D | hlo_dataflow_analysis_test.cc | 1466 HloInstruction::CreateReverse(vector_shape_, negate, {0})); in TEST_P()
2220 HloInstruction::CreateReverse(data_shape, neg, {0, 1})); in TEST_F()
2721 HloInstruction::CreateReverse(data_shape, operand, {0, 1})); in TEST_F()
2852 builder.AddInstruction(HloInstruction::CreateReverse(shape, param, {0})); in TEST_F()
|
| D | copy_insertion_test.cc | 558 HloInstruction::CreateReverse(data_shape_, gte11, {0})); in BuildNestedBodyComputation()
1787 HloInstruction::CreateReverse(element_shape, body_element_3, {0})); in TEST_F()
1940 HloInstruction::CreateReverse(element_shape, element_1, {0})); in MakeBenchmarkWhileBody()
|
| D | hlo_creation_utils.cc | 483 HloInstruction::CreateReverse(reverse_shape, operand, dimensions), in MakeReverseHlo()
|
| D | hlo_instruction.h | 1148 static std::unique_ptr<HloInstruction> CreateReverse(
|
| D | hlo_evaluator_test.cc | 1150 rhs_instruction = b.AddInstruction(HloInstruction::CreateReverse( in TEST_P()
3253 b.AddInstruction(HloInstruction::CreateReverse(shape, operand, {0, 1})); in TEST_P()
|
| D | hlo_instruction.cc | 290 CreateReverse(shape, operands(0), in CreateFromProto()
1428 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReverse( in CreateReverse() function in xla::HloInstruction
|
| D | buffer_assignment_test.cc | 1908 HloInstruction::CreateReverse(f32vec100_, log, {0})); in TEST_F()
|
| D | hlo_parser.cc | 2235 HloInstruction::CreateReverse(*shape, operands[0], *dimensions)); in CreateInstruction()
|
| D | algebraic_simplifier.cc | 4549 slice, HloInstruction::CreateReverse(new_slice->shape(), new_slice, in TryToReorderSliceAndReverse()
|
| D | algebraic_simplifier_test.cc | 4924 HloInstruction::CreateReverse(shape, a, /*dimensions=*/{2, 3})); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 468 auto reverse1 = builder.AddInstruction(HloInstruction::CreateReverse( in XLA_TEST_F()
484 auto reverse1 = builder.AddInstruction(HloInstruction::CreateReverse( in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 433 HloInstruction::CreateReverse(param_shape, param0, {0})); in TEST_F()
|