Home
last modified time | relevance | path

Searched refs:kPack (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_pack_to_reshape.cc33 if (pack_it->get()->type != OperatorType::kPack) { in Run()
Dresolve_constant_pack.cc58 if (base_op->type != OperatorType::kPack) { in Run()
Dgroup_bidirectional_sequence_ops.cc171 if (first_fw_sequence_op_input->type != OperatorType::kPack || in CheckTwoUnidirectionalSequenceOpsAreValid()
172 first_bw_sequence_op_input->type != OperatorType::kPack) { in CheckTwoUnidirectionalSequenceOpsAreValid()
Dpropagate_array_data_types.cc208 case OperatorType::kPack: { in Run()
Dquantize.cc66 type == OperatorType::kPack || type == OperatorType::kTopK_V2 || in SupportsQuantization()
Dpropagate_fixed_sizes.cc2244 case OperatorType::kPack: in Run()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h89 kPack, enumerator
1327 PackOperator() : Operator(OperatorType::kPack) {}
Dexport_tensorflow.cc2297 } else if (src_op.type == OperatorType::kPack) { in ConvertOperator()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc537 SerializeAndDeserialize(GetOperator("PACK", OperatorType::kPack), op); in TEST_F()
Doperator.cc2448 MakeUnique<Pack>(::tflite::BuiltinOperator_PACK, OperatorType::kPack)); in BuildOperatorList()