Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tests/
Dconv_node_name_test.py55 conv2d_op = layer(
60 _ = conv2d_op(input_tensor)
64 conv2d_op = layer(
69 _ = conv2d_op(input_tensor)
/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.cc396 tensorflow::NodeDef* conv2d_op = tensorflow_graph->add_node(); in ConvertConvOperator() local
397 conv2d_op->set_op("Conv2D"); in ConvertConvOperator()
398 conv2d_op->set_name(conv_output); in ConvertConvOperator()
399 *conv2d_op->add_input() = src_op.inputs[0]; in ConvertConvOperator()
400 *conv2d_op->add_input() = src_op.inputs[1]; in ConvertConvOperator()
401 (*conv2d_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertConvOperator()
408 auto& strides = (*conv2d_op->mutable_attr())["strides"]; in ConvertConvOperator()
415 auto& dilations = (*conv2d_op->mutable_attr())["dilations"]; in ConvertConvOperator()
429 (*conv2d_op->mutable_attr())["padding"].set_s(padding); in ConvertConvOperator()
549 tensorflow::NodeDef* conv2d_op = tensorflow_graph->add_node(); in ConvertTransposeConvOperator() local
[all …]