Searched refs:kTensorFlowReshape (Results 1 – 15 of 15) sorted by relevance
33 CHECK(op.type == OperatorType::kTensorFlowReshape); in IsReshapeTrivial()61 if (next_op->type == OperatorType::kTensorFlowReshape) { in IsReshapeTrivial()78 if (reshape_op->type != OperatorType::kTensorFlowReshape) { in Run()
31 if (reshape_op->type != OperatorType::kTensorFlowReshape) { in Run()
39 if (next_op->type == OperatorType::kTensorFlowReshape) { in Run()
43 unary_op->type != OperatorType::kTensorFlowReshape) { in Run()136 } else if (unary_op->type == OperatorType::kTensorFlowReshape) { in Run()
45 if (exchange_op->type != OperatorType::kTensorFlowReshape) { in Run()
100 if (previous_op && previous_op->type == OperatorType::kTensorFlowReshape) { in Run()
114 if (op_outputting_array->type == OperatorType::kTensorFlowReshape) { in DequantizeArray()
305 case OperatorType::kTensorFlowReshape: in Run()
49 type == OperatorType::kTensorFlowReshape || in SupportsQuantization()
1419 case OperatorType::kTensorFlowReshape: in Run()
107 kTensorFlowReshape, enumerator824 TensorFlowReshapeOperator() : Operator(OperatorType::kTensorFlowReshape) {}
683 providing_op->type == OperatorType::kTensorFlowReshape) { in ConvertSoftmaxOperator()723 if (providing_op->type == OperatorType::kTensorFlowReshape) { in ConvertLogSoftmaxOperator()1672 } else if (src_op.type == OperatorType::kTensorFlowReshape) { in ConvertOperator()
340 case OperatorType::kTensorFlowReshape: in OperatorSupportsFusedActivation()
236 GetOperator("RESHAPE", OperatorType::kTensorFlowReshape), op); in TEST_F()
807 OperatorType::kTensorFlowReshape)); in BuildOperatorList()