HomeSort by relevance Sort by last modified time
    Searched refs:argmax_op (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
propagate_array_data_types.cc 113 auto* argmax_op = static_cast<ArgMaxOperator*>(op); local
114 model->GetArray(op->outputs[0]).data_type = argmax_op->output_data_type;
  /external/tensorflow/tensorflow/python/kernel_tests/
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)
    [all...]
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 1249 tensorflow::NodeDef* argmax_op = tensorflow_graph->add_node(); local
    [all...]

Completed in 125 milliseconds