Searched refs:argmax_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_array_data_types.cc | 113 auto* argmax_op = static_cast<ArgMaxOperator*>(op); in Run() local
114 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 | 788 _, argmax_op = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
789 argmax = self.evaluate(argmax_op)
817 _, argmax_op = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
818 argmax = self.evaluate(argmax_op)
856 out_op, argmax_op = nn_ops.max_pool_with_argmax(
863 out, argmax = self.evaluate([out_op, argmax_op])
865 self.assertShapeEqual(argmax, argmax_op)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1249 tensorflow::NodeDef* argmax_op = tensorflow_graph->add_node(); in ConvertArgMaxOperator() local
1250 argmax_op->set_op("ArgMax"); in ConvertArgMaxOperator()
1251 argmax_op->set_name(src_op.outputs[0]); in ConvertArgMaxOperator()
1253 *argmax_op->add_input() = src_op.inputs[0]; in ConvertArgMaxOperator()
1254 *argmax_op->add_input() = src_op.inputs[1]; in ConvertArgMaxOperator()
1255 (*argmax_op->mutable_attr())["T"].set_type( in ConvertArgMaxOperator()
1257 (*argmax_op->mutable_attr())["Tidx"].set_type( in ConvertArgMaxOperator()
1259 (*argmax_op->mutable_attr())["output_type"].set_type( in ConvertArgMaxOperator()
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 9 tensorflow/core/kernels/argmax_op.cc
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3257 ":argmax_op",
3286 name = "argmax_op",
3287 prefix = "argmax_op",
5588 "argmax_op.h",
5653 "argmax_op.cc",
|