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

  /external/tensorflow/tensorflow/python/ops/
conv2d_benchmark.py 72 conv2d_op = nn_ops.conv2d(
74 outputs.append(conv2d_op)
76 with ops.control_dependencies([conv2d_op]):
77 conv2d_op = nn_ops.conv2d(
79 outputs.append(conv2d_op)
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 392 tensorflow::NodeDef* conv2d_op = tensorflow_graph->add_node(); local
393 conv2d_op->set_op("Conv2D");
394 conv2d_op->set_name(conv_output);
395 *conv2d_op->add_input() = src_op.inputs[0];
396 *conv2d_op->add_input() = src_op.inputs[1];
397 (*conv2d_op->mutable_attr())["T"].set_type(DT_FLOAT);
404 auto& strides = (*conv2d_op->mutable_attr())["strides"];
411 auto& dilations = (*conv2d_op->mutable_attr())["dilations"];
425 (*conv2d_op->mutable_attr())["padding"].set_s(padding);
544 tensorflow::NodeDef* conv2d_op = tensorflow_graph->add_node() local
    [all...]

Completed in 112 milliseconds