| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | remove_trivial_reshape.cc | 33 CHECK(op.type == OperatorType::kReshape); in IsReshapeTrivial()
61 if (next_op->type == OperatorType::kReshape) { in IsReshapeTrivial()
90 if (reshape_op->type != OperatorType::kReshape) { in Run()
|
| D | resolve_constant_unary.cc | 133 case OperatorType::kReshape: in Run()
177 if (unary_op->type == OperatorType::kReshape) { in Run()
238 } else if (unary_op->type == OperatorType::kReshape) { in Run()
|
| D | propagate_fake_quant_num_bits.cc | 108 case OperatorType::kReshape: in DoesOpBlockBackwardPropagation()
136 case OperatorType::kReshape: in DoesOpInputBlockBackwardPropagation()
|
| D | resolve_squeeze_attributes.cc | 42 if (next_op->type == OperatorType::kReshape) { in Run()
|
| D | resolve_reshape_attributes.cc | 34 if (reshape_op->type != OperatorType::kReshape) { in Run()
|
| D | resolve_constant_reshape.cc | 31 if (base_op->type != OperatorType::kReshape) { in Run()
|
| D | reorder_elementwise_unary.cc | 55 case OperatorType::kReshape: in IsMoveOperator()
|
| D | remove_trivial_passthrough.cc | 120 if (passthru_op->type != OperatorType::kReshape && in RemoveTrivialPassthroughOp()
|
| D | merge_reshape_into_preceding_transpose.cc | 110 it->get(), OperatorType::kReshape); in Run()
|
| D | move_binary_operator_before_reshape.cc | 122 if (reshape_op->type != OperatorType::kReshape) { in Run()
|
| D | resolve_tensorflow_matmul.cc | 167 if (previous_op && previous_op->type == OperatorType::kReshape) { in Run()
|
| D | dequantize.cc | 114 if (op_outputting_array->type == OperatorType::kReshape) { in DequantizeArray()
|
| D | reorder_reshape_transpose.cc | 130 OperatorType::kReshape); in Run()
|
| D | quantize.cc | 51 type == OperatorType::kPadV2 || type == OperatorType::kReshape || in SupportsQuantization()
364 op.type == OperatorType::kReshape || op.type == OperatorType::kSplit || in ChooseQuantizationForOperatorOutput()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover.cc | 49 return instruction->opcode() == HloOpcode::kReshape || in IsReshapeOrTranspose()
125 case HloOpcode::kReshape: in AreEquivalentReshapes()
147 if (first_reshape_operand->opcode() == HloOpcode::kReshape) { in UpdateOperand()
166 case HloOpcode::kReshape: in UpdateOperand()
240 case HloOpcode::kReshape: in PerformSinkReshapeOrTranspose()
|
| D | bfloat16_support.cc | 94 case HloOpcode::kReshape: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | ar_crs_combiner.cc | 50 case HloOpcode::kReshape: in MatchesArCrsPattern()
335 case HloOpcode::kReshape: in RewriteGraph()
|
| D | dynamic_padder.cc | 76 case HloOpcode::kReshape: in ChooseIdentityValue()
|
| D | hlo_opcode.h | 120 V(kReshape, "reshape", 1) \
|
| D | while_loop_invariant_code_motion.cc | 116 case HloOpcode::kReshape: in NotWorthHoistingIndividually()
|
| D | hlo_instruction.cc | 1181 absl::WrapUnique(new HloInstruction(HloOpcode::kReshape, shape)); in CreateReshape()
1460 case HloOpcode::kReshape: in CloneWithNewOperands()
1760 case HloOpcode::kReshape: in IdenticalSlowPath()
2433 opcode() == HloOpcode::kReshape) { in ToCategory()
2609 case HloOpcode::kReshape: in Visit()
2931 case HloOpcode::kReshape: in OperandElementUse()
2972 if (HloOpcode::kReshape != opcode_) { in ReshapeMerelyInsertsOrDeletes1SizedDimensions()
3144 case HloOpcode::kReshape: in CouldBeBitcast()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 40 hlo.opcode() == HloOpcode::kReshape || in CanBeLoopFused()
|
| D | cpu_instruction_fusion_test.cc | 328 module.get(), {HloOpcode::kNegate, HloOpcode::kReshape, HloOpcode::kExp, in TEST_F()
360 {HloOpcode::kTanh, HloOpcode::kDynamicSlice, HloOpcode::kReshape, in TEST_F()
440 {HloOpcode::kNegate, HloOpcode::kReshape, HloOpcode::kParameter}); in TEST_F()
625 HloOpcode::kDynamicUpdateSlice, HloOpcode::kReshape, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 177 instr.opcode() == HloOpcode::kReshape || in IsLoopFusible()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 210 EXPECT_EQ(HloOpcode::kReshape, in TEST_F()
|