Searched refs:casted_input (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 197 Output casted_input = Cast(root.WithOpName("casted_input"), input, dtype); in CompareFusedAndSeparate() local
223 root.WithOpName("fused_conv"), casted_input, resize_size, paddings, in CompareFusedAndSeparate()
258 Output casted_input = Cast(root.WithOpName("casted_input"), input, dtype); in CompareFusedPadOnlyAndSeparate() local
271 Output mirror_pad = MirrorPad(root.WithOpName("mirror_pad"), casted_input, in CompareFusedPadOnlyAndSeparate()
277 root.WithOpName("fused_conv"), casted_input, paddings, casted_filter, in CompareFusedPadOnlyAndSeparate()
|
| D | depthwise_conv_grad_op.cc | 1187 Tensor casted_input = input; in Compute() local
1198 &casted_input)); in Compute()
1199 cast(device, casted_input.template flat<float>(), in Compute()
1207 auto input_ptr = casted_input.template flat<U>().data(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3659 auto casted_input = in matchAndRewrite() local
3666 loc, casted_input.getResult(), init, in matchAndRewrite()
|