Home
last modified time | relevance | path

Searched refs:range_op (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_array_data_types.cc130 auto* range_op = static_cast<RangeOperator*>(op); in Run() local
133 if (range_op->dtype != ArrayDataType::kNone) { in Run()
135 data_type = range_op->dtype; in Run()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1335 tensorflow::NodeDef* range_op = tensorflow_graph->add_node(); in ConvertRangeOperator() local
1336 range_op->set_op("Range"); in ConvertRangeOperator()
1337 range_op->set_name(src_op.outputs[0]); in ConvertRangeOperator()
1339 *range_op->add_input() = src_op.inputs[0]; in ConvertRangeOperator()
1340 *range_op->add_input() = src_op.inputs[1]; in ConvertRangeOperator()
1341 *range_op->add_input() = src_op.inputs[2]; in ConvertRangeOperator()
1342 (*range_op->mutable_attr())["Tidx"].set_type( in ConvertRangeOperator()