| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_test.py | 138 weights_op = constant_op.constant(
141 input_op, weights_op, [1, 1, 1, 1], padding="SAME", name="conv_op")
190 weights_op = constant_op.constant(
194 weights_op = constant_op.constant(
198 weights_op, [1, 1, 1, 1],
250 weights_op = constant_op.constant(
254 weights_op = constant_op.constant(
267 weights_op, [1, 1, 1, 1],
307 weights_op = constant_op.constant(
310 pad_op, weights_op, [1, 1, 1, 1], padding="VALID", name="output")
[all …]
|
| D | optimize_for_inference_lib.py | 269 weights_op = node_from_map(input_node_map, conv_op.input[1])
270 if weights_op.op != "Const":
273 " run first?" % (conv_op.name, weights_op))
275 weights = values_from_const(weights_op)
343 nodes_to_skip[weights_op.name] = True
374 scaled_weights_op.attr["dtype"].CopyFrom(weights_op.attr["dtype"])
380 if weights_node == weights_op.name:
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 53 Output weights_op = in TestFoldBatchNormsConv2D() local
56 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2D()
104 Output weights_op = in TestFoldBatchNormsDepthwiseConv2dNative() local
108 weights_op, {1, 1, 1, 1}, "VALID"); in TestFoldBatchNormsDepthwiseConv2dNative()
155 Output weights_op = in TestFoldBatchNormsConv2DShared() local
158 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2DShared()
212 Output weights_op = in TestFoldBatchNormsMatMul() local
216 MatMul(root.WithOpName("matmul_op"), input_op, weights_op); in TestFoldBatchNormsMatMul()
|
| D | fuse_convolutions_test.cc | 69 Output weights_op = in TestFuseResizePadAndConv() local
72 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, in TestFuseResizePadAndConv()
119 Output weights_op = in TestFuseResizeAndConv() local
122 Output conv_op = Conv2D(root.WithOpName("output"), resize_op, weights_op, in TestFuseResizeAndConv()
171 Output weights_op = in TestFusePadAndConv() local
174 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, in TestFusePadAndConv()
|
| D | fold_old_batch_norms_test.cc | 54 Output weights_op = in TestFoldOldBatchNorms() local
57 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldOldBatchNorms()
138 Output weights_op = in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
142 weights_op, {1, 1, 1, 1}, "VALID"); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
216 Output weights_op = in TestFoldFusedBatchNorms() local
219 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldFusedBatchNorms()
293 Output weights_op = in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
297 weights_op, {1, 1, 1, 1}, "VALID"); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative()
472 Output weights_op = in TestFoldFusedBatchNormsWithBatchToSpace() local
475 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldFusedBatchNormsWithBatchToSpace()
|
| D | round_weights_test.cc | 56 Output weights_op = in TestRoundWeights() local
59 Output conv_op = Conv2D(root.WithOpName("output"), input_op, weights_op, in TestRoundWeights()
|
| D | flatten_atrous_test.cc | 67 Output weights_op = in TestFlattenAtrousConv() local
71 weights_op, {1, 1, 1, 1}, "VALID"); in TestFlattenAtrousConv()
|
| D | quantize_weights_test.cc | 52 Output weights_op = ops::Const(root.WithOpName("weights_op"), in BuildGraphDef() local
56 weights_op, {1, 1, 1, 1}, "VALID"); in BuildGraphDef()
|