Home
last modified time | relevance | path

Searched refs:set_op (Results 1 – 25 of 190) sorted by relevance

12345678

/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns_test.cc627 auto set_op = [&op](const std::vector<int32>& strides, const string& padding, in TEST() local
647 set_op({{1, 1, 0, 1}}, "VALID", "NHWC", "HWIO"); in TEST()
651 set_op({{1, 1, 1, 1}}, "VALID", "NHWC", "HWIO"); in TEST()
655 set_op({{1, 1, 1, 1}}, "VALID", "NHWC", "HWIO"); in TEST()
659 set_op({{1, 2, 2, 1}}, "VALID", "NHWC", "HWIO"); in TEST()
663 set_op({{1, 2, 1, 1}}, "VALID", "NHWC", "HWIO"); in TEST()
667 set_op({{1, 1, 2, 1}}, "VALID", "NHWC", "HWIO"); in TEST()
685 set_op({{1, 1, 1, 1}}, "VALID", "NCHW", "HWIO"); in TEST()
689 set_op({{1, 1, 1, 1}}, "VALID", "NCHW", "HWIO"); in TEST()
693 set_op({{1, 1, 2, 2}}, "VALID", "NCHW", "HWIO"); in TEST()
[all …]
Dgraph_def_util_test.cc294 function_def->add_node_def()->set_op(op); in TEST()
296 graph_def.add_node()->set_op("F"); in TEST()
302 node->set_op(op); in TEST()
341 b->add_node_def()->set_op("A"); in TEST()
342 c->add_node_def()->set_op("B"); in TEST()
344 b->add_node_def()->set_op("B"); in TEST()
345 c->add_node_def()->set_op("C"); in TEST()
349 graph_def.add_node()->set_op("C"); in TEST()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsort_by_execution_order_test.cc44 add_node->set_op("Add"); in TestSimpleAdd()
50 b_node->set_op("Const"); in TestSimpleAdd()
54 a_node->set_op("Const"); in TestSimpleAdd()
71 negative_node->set_op("Negative"); in TestSimpleLinear()
76 relu_node->set_op("Relu"); in TestSimpleLinear()
81 sqrt_node->set_op("Sqrt"); in TestSimpleLinear()
86 const_node->set_op("Const"); in TestSimpleLinear()
104 add_node1->set_op("Add"); in TestSimpleTree()
110 add_node2->set_op("Add"); in TestSimpleTree()
116 add_node3->set_op("Add"); in TestSimpleTree()
[all …]
Dstrip_unused_nodes_test.cc43 add_node->set_op("Add"); in TestSimpleAdd()
49 a_node->set_op("Const"); in TestSimpleAdd()
53 b_node->set_op("Const"); in TestSimpleAdd()
57 c_node->set_op("Const"); in TestSimpleAdd()
75 add_node1->set_op("Add"); in TestCommonAncestor()
81 add_node2->set_op("Add"); in TestCommonAncestor()
87 add_node3->set_op("Add"); in TestCommonAncestor()
93 const_node1->set_op("Const"); in TestCommonAncestor()
97 const_node2->set_op("Const"); in TestCommonAncestor()
101 const_node3->set_op("Const"); in TestCommonAncestor()
[all …]
Dremove_nodes_test.cc43 add_node1->set_op("Add"); in TestRemoveNodes()
49 add_node2->set_op("Add"); in TestRemoveNodes()
55 add_node3->set_op("Add"); in TestRemoveNodes()
61 identity_node1->set_op("Identity"); in TestRemoveNodes()
66 identity_node2->set_op("Identity"); in TestRemoveNodes()
71 identity_node3->set_op("Identity"); in TestRemoveNodes()
76 const_node1->set_op("Const"); in TestRemoveNodes()
80 const_node2->set_op("Const"); in TestRemoveNodes()
84 const_node3->set_op("Const"); in TestRemoveNodes()
88 add_node4->set_op("Add"); in TestRemoveNodes()
[all …]
Dbackports_test.cc47 mul_node1->set_op("Mul"); in TestBackportConcatV2()
53 add_node2->set_op("Add"); in TestBackportConcatV2()
59 add_node3->set_op("Add"); in TestBackportConcatV2()
65 const_node1->set_op("Const"); in TestBackportConcatV2()
69 const_node2->set_op("Const"); in TestBackportConcatV2()
73 const_node3->set_op("Const"); in TestBackportConcatV2()
77 concat_node->set_op("ConcatV2"); in TestBackportConcatV2()
113 size_node->set_op("Const"); in TEST()
120 tensor_array_node->set_op("TensorArrayV3"); in TEST()
130 handle_output_node->set_op("Identity"); in TEST()
[all …]
Dobfuscate_names_test.cc43 add_node1->set_op("Add"); in TestSimpleTree()
49 add_node2->set_op("Add"); in TestSimpleTree()
55 add_node3->set_op("Add"); in TestSimpleTree()
61 const_node1->set_op("Const"); in TestSimpleTree()
65 const_node2->set_op("Const"); in TestSimpleTree()
69 const_node3->set_op("Const"); in TestSimpleTree()
73 const_node4->set_op("Const"); in TestSimpleTree()
96 const_node->set_op("Const"); in TestManyNodes()
115 const_node->set_op("Const"); in TestNameClashes()
Dremove_device_test.cc43 mul_node1->set_op("Mul"); in TestRemoveDevice()
50 add_node2->set_op("Add"); in TestRemoveDevice()
57 add_node3->set_op("Add"); in TestRemoveDevice()
63 const_node1->set_op("Const"); in TestRemoveDevice()
67 const_node2->set_op("Const"); in TestRemoveDevice()
71 const_node3->set_op("Const"); in TestRemoveDevice()
75 add_node4->set_op("Add"); in TestRemoveDevice()
Drename_op_test.cc43 mul_node1->set_op("Mul"); in TestRenameOp()
49 add_node2->set_op("Add"); in TestRenameOp()
55 add_node3->set_op("Add"); in TestRenameOp()
61 const_node1->set_op("Const"); in TestRenameOp()
65 const_node2->set_op("Const"); in TestRenameOp()
69 const_node3->set_op("Const"); in TestRenameOp()
73 add_node4->set_op("Add"); in TestRenameOp()
Dset_device_test.cc42 mul_node1->set_op("Mul"); in CreateDeviceGraph()
49 add_node2->set_op("Add"); in CreateDeviceGraph()
56 add_node3->set_op("Add"); in CreateDeviceGraph()
62 const_node1->set_op("Const"); in CreateDeviceGraph()
66 const_node2->set_op("Const"); in CreateDeviceGraph()
70 const_node3->set_op("Const"); in CreateDeviceGraph()
74 add_node4->set_op("Add"); in CreateDeviceGraph()
Dremove_attribute_test.cc43 mul_node1->set_op("Mul"); in TestRemoveAttribute()
51 add_node2->set_op("Add"); in TestRemoveAttribute()
60 add_node3->set_op("Add"); in TestRemoveAttribute()
66 const_node1->set_op("Const"); in TestRemoveAttribute()
70 const_node2->set_op("Const"); in TestRemoveAttribute()
74 const_node3->set_op("Const"); in TestRemoveAttribute()
78 add_node4->set_op("Add"); in TestRemoveAttribute()
Drename_node_test.cc37 node->set_op("Placeholder"); in TEST()
41 node_splitter->set_op("Split"); in TEST()
44 node_adder->set_op("Add"); in TEST()
72 node->set_op("Placeholder"); in TEST()
76 node_splitter->set_op("Split"); in TEST()
79 node_adder->set_op("Add"); in TEST()
Drename_attribute_test.cc43 mul_node1->set_op("Mul"); in TestRenameAttribute()
51 add_node2->set_op("Add"); in TestRenameAttribute()
60 add_node3->set_op("Add"); in TestRenameAttribute()
66 const_node1->set_op("Const"); in TestRenameAttribute()
70 const_node2->set_op("Const"); in TestRenameAttribute()
74 const_node3->set_op("Const"); in TestRenameAttribute()
78 add_node4->set_op("Add"); in TestRenameAttribute()
Dbackports.cc39 concat_node.set_op("Concat"); in BackportConcatV2Transform()
80 tensor_array_v2_node.set_op("TensorArrayV2"); in BackportTensorArrayV3Transform()
82 tensor_array_v2_node.set_op("TensorArrayGradV2"); in BackportTensorArrayV3Transform()
88 replacement_flow_node.set_op("Const"); in BackportTensorArrayV3Transform()
122 v2_node.set_op(v3_node.op().substr(0, v3_node.op().size() - 1) + "2"); in BackportTensorArrayV3Transform()
Dquantize_nodes.cc340 min_node.set_op("Const"); in QuantizePlaceholders()
349 max_node.set_op("Const"); in QuantizePlaceholders()
359 dequantize_node.set_op("Dequantize"); in QuantizePlaceholders()
418 quantize_node.set_op("QuantizeV2"); in ConvertFakeQuantsToRequantize()
428 requantize_node.set_op("Requantize"); in ConvertFakeQuantsToRequantize()
441 dequantize_node.set_op("Dequantize"); in ConvertFakeQuantsToRequantize()
748 reshape_dims.set_op("Const"); in QuantizeNodes()
758 reduction_dims.set_op("Const"); in QuantizeNodes()
769 reshape_node.set_op("Reshape"); in QuantizeNodes()
777 min_node.set_op("Min"); in QuantizeNodes()
[all …]
Dquantize_weights.cc100 quantized_const_node.set_op("Const"); in QuantizeWeights()
109 min_node.set_op("Const"); in QuantizeWeights()
118 max_node.set_op("Const"); in QuantizeWeights()
127 dequantize_node.set_op("Dequantize"); in QuantizeWeights()
Dfuse_convolutions.cc61 fused_conv.set_op("FusedResizeAndPadConv2D"); in FuseResizePadAndConv()
107 pad_dims_node.set_op("Const"); in FuseResizeAndConv()
116 fused_conv.set_op("FusedResizeAndPadConv2D"); in FuseResizeAndConv()
173 fused_conv.set_op("FusedPadConv2D"); in FusePadAndConv()
/external/tensorflow/tensorflow/core/ops/
Dnn_ops_test.cc184 auto set_op = [&op](bool is_training, float exponential_avg_factor, in TEST() local
198 set_op(true, 1.0, "NHWC"); in TEST()
212 set_op(true, 0.5, "NHWC"); in TEST()
226 set_op(true, 1.0, "NCHW"); in TEST()
240 set_op(false, 1.0, "NHWC"); in TEST()
258 set_op(false, 1.0, "NCHW"); in TEST()
279 auto set_op = [&op](string data_format) { in TEST() local
290 set_op("NCHW"); in TEST()
306 set_op("NHWC"); in TEST()
493 auto set_op = [&op](const std::vector<int32>& strides, in TEST() local
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dtpu_test.cc32 tpu_graph.add_node()->set_op("Add"); in TEST_F()
35 function_def->add_node_def()->set_op("Mul"); in TEST_F()
44 tpu_graph.add_node()->set_op("TPUPartitionedCall"); in TEST_F()
53 tpu_graph.add_node()->set_op("BatchFunction"); in TEST_F()
56 function_def->add_node_def()->set_op("TPUPartitionedCall"); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc163 const_op->set_op("Const"); in ConvertFloatTensorConst()
205 const_op->set_op("Const"); in ConvertFloatTensorConst()
228 const_op->set_op("Const"); in ConvertBoolTensorConst()
252 const_op->set_op("Const"); in ConvertIntTensorConst()
276 const_op->set_op("Const"); in CreateIntTensorConst()
301 const_op->set_op("Const"); in ConvertComplex64TensorConst()
324 const_op->set_op("Const"); in CreateMatrixShapeTensorConst()
342 const_op->set_op("Const"); in CreateDummyConcatDimTensorConst()
357 const_op->set_op("Const"); in CreateReshapeShapeTensorConst()
397 conv2d_op->set_op("Conv2D"); in ConvertConvOperator()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer_test.cc39 node.set_op("Conv2D"); in TEST()
69 node.set_op("Conv2D"); in TEST()
99 node.set_op("Conv2D"); in TEST()
129 node.set_op("Conv2D"); in TEST()
151 node.set_op("Conv2D"); in TEST()
162 node.set_op("Conv2D"); in GetDefaultDeviceName()
211 node.set_op("Conv2D"); in TEST()
261 node.set_op("Conv2D"); in TEST()
277 node.set_op("Conv2D"); in TEST()
297 node.set_op("Conv2D"); in TEST()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc259 shp->set_op("Const"); in RunInfiniteTFLoop()
268 r->set_op("RandomUniform"); in RunInfiniteTFLoop()
275 e->set_op("Enter"); in RunInfiniteTFLoop()
282 m->set_op("Merge"); in RunInfiniteTFLoop()
290 t->set_op("Const"); in RunInfiniteTFLoop()
300 c->set_op("LoopCond"); in RunInfiniteTFLoop()
306 s->set_op("Switch"); in RunInfiniteTFLoop()
312 i->set_op("Identity"); in RunInfiniteTFLoop()
318 n->set_op("NextIteration"); in RunInfiniteTFLoop()
324 x->set_op("Exit"); in RunInfiniteTFLoop()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc46 x->set_op("Placeholder"); in SumGraph()
50 y->set_op("Placeholder"); in SumGraph()
54 sum->set_op("Add"); in SumGraph()
107 unused->set_op("Placeholder"); in TEST()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc61 config_def.set_op(kInternalConfigureOp); in AddConfigurationNode()
83 host_config_def.set_op(kHostConfigureOp); in AddHostConfigNode()
104 wait_def.set_op(kWaitOp); in AddWaitNode()
131 global_tpu_array_def.set_op(kGlobalTPUArrayOp); in AddGlobalTPUArrayNode()
153 sync_def.set_op(kIdentityOp); in AddSynchronizationNode()
190 shutdown_def.set_op(kInternalShutdownOp); in AddShutdownNode()
217 host_disconnect_def.set_op(kHostDisconnectOp); in AddHostDisconnectNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1080 fused_op.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
1083 fused_op.set_op(kFusedDepthwiseConv2dNative); in IsCpuCompatibleDataType()
1086 fused_op.set_op(kFusedMatMul); in IsCpuCompatibleDataType()
1128 fused_op.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
1132 fused_op.set_op(kFusedDepthwiseConv2dNative); in IsCpuCompatibleDataType()
1135 fused_op.set_op(kFusedMatMul); in IsCpuCompatibleDataType()
1174 fused_conv2d.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
1216 fused_conv2d.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
1259 fused_conv2d.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
1312 contraction_node.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
[all …]

12345678