Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dconv2d_benchmark.py72 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/
Dexport_tensorflow.cc392 tensorflow::NodeDef* conv2d_op = tensorflow_graph->add_node(); in ConvertConvOperator() local
393 conv2d_op->set_op("Conv2D"); in ConvertConvOperator()
394 conv2d_op->set_name(conv_output); in ConvertConvOperator()
395 *conv2d_op->add_input() = src_op.inputs[0]; in ConvertConvOperator()
396 *conv2d_op->add_input() = src_op.inputs[1]; in ConvertConvOperator()
397 (*conv2d_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertConvOperator()
404 auto& strides = (*conv2d_op->mutable_attr())["strides"]; in ConvertConvOperator()
411 auto& dilations = (*conv2d_op->mutable_attr())["dilations"]; in ConvertConvOperator()
425 (*conv2d_op->mutable_attr())["padding"].set_s(padding); in ConvertConvOperator()
544 tensorflow::NodeDef* conv2d_op = tensorflow_graph->add_node(); in ConvertTransposeConvOperator() local
[all …]