Searched refs:kSplit (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_lstm.cc | 207 if (!MatchOperatorInputs(*state_info_tanh, *model, OperatorType::kSplit, in Run()
213 if (!MatchOperatorInputs(*state_remember_sig, *model, OperatorType::kSplit, in Run()
219 if (!MatchOperatorInputs(*state_forget_sig, *model, OperatorType::kSplit, in Run()
225 if (!MatchOperatorInputs(*fc_output_sig, *model, OperatorType::kSplit, in Run()
|
| D | quantize.cc | 99 OperatorType::kSplit, in SupportsQuantization()
412 op.type == OperatorType::kReshape || op.type == OperatorType::kSplit || in ChooseQuantizationForOperatorOutput()
417 if (op.type == OperatorType::kSplit) { in ChooseQuantizationForOperatorOutput()
|
| D | propagate_array_data_types.cc | 85 case OperatorType::kSplit: in Run()
|
| D | hardcode_min_max.cc | 455 case OperatorType::kSplit: in Run()
|
| D | propagate_fixed_sizes.cc | 2296 case OperatorType::kSplit: in Run()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version.cc | 156 {{OperatorType::kSplit, 1}, "1.5.0"}, in GetMinimumRuntimeVersionForModel()
157 {{OperatorType::kSplit, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
158 {{OperatorType::kSplit, 3}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
159 {{OperatorType::kSplit, 4}, kPendingReleaseOpVersion}, in GetMinimumRuntimeVersionForModel()
|
| D | operator_test.cc | 335 SerializeAndDeserialize(GetOperator("SPLIT", OperatorType::kSplit), op); in TEST_F()
|
| D | operator.cc | 1919 MakeUnique<Split>(::tflite::BuiltinOperator_SPLIT, OperatorType::kSplit)); in BuildOperatorList()
|
| /external/swiftshader/third_party/subzero/src/ |
| D | IceTargetLoweringX8632Traits.h | 757 enum OperandKindX8632 { k__Start = ::Ice::Operand::kTarget, kMem, kSplit };
855 return Operand->getKind() == static_cast<OperandKind>(kSplit);
860 : X86Operand(kSplit, IceType_i32), Var(Var), Part(Part) {
|
| D | IceTargetLoweringX8664Traits.h | 851 enum OperandKindX8664 { k__Start = ::Ice::Operand::kTarget, kMem, kSplit };
943 return Operand->getKind() == static_cast<OperandKind>(kSplit);
948 : X86Operand(kSplit, IceType_i32), Var(Var), Part(Part) {
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 128 kSplit, enumerator
1502 TensorFlowSplitOperator() : Operator(OperatorType::kSplit) {}
|
| D | export_tensorflow.cc | 2227 } else if (src_op.type == OperatorType::kSplit) { in ConvertOperator()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 64 constexpr char kSplit[] = "Split"; variable
490 device_cost_impl_.emplace(kSplit, in OpLevelCostEstimator()
|