Searched refs:argmax_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_array_data_types.cc | 118 auto* argmax_op = static_cast<ArgMaxOperator*>(op); in Run() local
119 model->GetArray(op->outputs[0]).data_type = argmax_op->output_data_type; in Run()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 947 _, argmax_op = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
948 argmax = self.evaluate(argmax_op)
979 _, argmax_op = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
980 argmax = self.evaluate(argmax_op)
1020 out_op, argmax_op = nn_ops.max_pool_with_argmax(
1027 out, argmax = self.evaluate([out_op, argmax_op])
1029 self.assertShapeEqual(argmax, argmax_op)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1262 tensorflow::NodeDef* argmax_op = tensorflow_graph->add_node(); in ConvertArgMaxOperator() local
1263 argmax_op->set_op("ArgMax"); in ConvertArgMaxOperator()
1264 argmax_op->set_name(src_op.outputs[0]); in ConvertArgMaxOperator()
1266 *argmax_op->add_input() = src_op.inputs[0]; in ConvertArgMaxOperator()
1267 *argmax_op->add_input() = src_op.inputs[1]; in ConvertArgMaxOperator()
1268 (*argmax_op->mutable_attr())["T"].set_type( in ConvertArgMaxOperator()
1270 (*argmax_op->mutable_attr())["Tidx"].set_type( in ConvertArgMaxOperator()
1272 (*argmax_op->mutable_attr())["output_type"].set_type( in ConvertArgMaxOperator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3256 ":argmax_op",
3286 name = "argmax_op",
3287 prefix = "argmax_op",
5896 "argmax_op.h",
5992 "argmax_op.cc",
|