| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cudnn_deterministic_base.py | 82 out_op = self._random_data_op(out_shape)
83 return out_op
106 out_op = nn_ops.conv3d(
113 self._assert_reproducible(out_op)
124 out_op = self._random_out_op(in_shape, filter_shape, strides, padding)
126 in_op, filter_shape, out_op, strides=strides, padding=padding)
138 out_op = self._random_out_op(in_shape, filter_shape, strides, padding)
140 in_shape, filter_op, out_op, strides=strides, padding=padding)
|
| D | pooling_ops_test.py | 927 out_op, _ = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
928 gpu_val = self.evaluate(out_op)
931 out_op = nn_ops.max_pool(t, ksize, strides, padding)
932 cpu_val = self.evaluate(out_op)
950 out_op = gen_nn_ops.max_pool_grad_with_argmax(t, grad_in, argmax, ksize,
952 gpu_val = self.evaluate(out_op)
953 self.assertShapeEqual(gpu_val, out_op)
956 out_op = nn_ops.max_pool(t, ksize, strides, padding)
957 orig_out = self.evaluate(out_op)
959 out_op = gen_nn_ops.max_pool_grad(t, orig_out, grad_in, ksize, strides,
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_op_rewrite_registry_test.cc | 29 std::unique_ptr<tensorflow::EagerOperation>* out_op) override { in Run() argument
35 out_op->reset(op); in Run()
53 std::unique_ptr<tensorflow::EagerOperation> out_op; in TEST() local
56 EagerOpRewriteRegistry::PRE_EXECUTION, &orig_op, &out_op)); in TEST()
58 EXPECT_EQ("NoOp", out_op->Name()); in TEST()
|
| D | mkl_eager_op_rewrite_test.cc | 58 std::unique_ptr<tensorflow::EagerOperation> out_op; in CheckRewrite() local
61 EagerOpRewriteRegistry::PRE_EXECUTION, orig_op, &out_op)); in CheckRewrite()
65 if (out_op) { in CheckRewrite()
66 actual_op_name = out_op->Name(); in CheckRewrite()
|
| D | eager_op_rewrite_registry.cc | 40 std::unique_ptr<tensorflow::EagerOperation>* out_op) { in RunRewrite() argument
43 TF_RETURN_IF_ERROR(rewrite->Run(orig_op, out_op)); in RunRewrite()
|
| D | eager_op_rewrite_registry.h | 39 std::unique_ptr<tensorflow::EagerOperation>* out_op) = 0;
68 std::unique_ptr<tensorflow::EagerOperation>* out_op);
|
| D | mkl_eager_op_rewrite.cc | 42 std::unique_ptr<tensorflow::EagerOperation>* out_op);
128 std::unique_ptr<tensorflow::EagerOperation>* out_op) { in Run() argument
130 TF_CHECK_OK(RewriteToMklOp(orig_op, out_op)); in Run()
|
| D | execute.cc | 713 std::unique_ptr<tensorflow::EagerOperation> out_op; in EagerLocalExecute() local
715 EagerOpRewriteRegistry::POST_PLACEMENT, op, &out_op)); in EagerLocalExecute()
716 if (out_op) { in EagerLocalExecute()
717 op = out_op.get(); in EagerLocalExecute()
1083 std::unique_ptr<tensorflow::EagerOperation> out_op; in EagerExecute() local
1085 EagerOpRewriteRegistry::PRE_EXECUTION, op, &out_op)); in EagerExecute()
1088 if (out_op) { in EagerExecute()
1089 op = out_op.get(); in EagerExecute()
1099 if (out_op) { in EagerExecute()
1100 op = out_op.get(); in EagerExecute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | node_expansion_pass.cc | 37 std::unique_ptr<EagerOperation>* out_op) { in Run() argument
86 out_op->reset(new_op); in Run()
|
| D | node_expansion_pass.h | 35 std::unique_ptr<tensorflow::EagerOperation>* out_op) override;
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 2984 out_op = image_ops.resize_images_v2(
2986 gpu_val = self.evaluate(out_op)
2990 out_op = image_ops.resize_images_v2(
2992 cpu_val = self.evaluate(out_op)
3027 out_op = image_ops.resize_images(image, new_size,
3029 value[use_gpu] = self.evaluate(out_op)
3539 out_op = image_ops.resize_images(
3544 gpu_val = self.evaluate(out_op)
3548 out_op = image_ops.resize_images(
3553 cpu_val = self.evaluate(out_op)
[all …]
|