| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 35 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
57 TensorShape x_shape({1, 2, 3}); in TEST_F() local
67 TensorShape x_shape({1, 2, 3}); in TEST_F() local
77 TensorShape x_shape({4, 2, 3}); in TEST_F() local
86 TensorShape x_shape({4, 2, 3}); in TEST_F() local
102 TensorShape x_shape({5, 2}); in TEST_F() local
112 TensorShape x_shape({}); in TEST_F() local
120 TensorShape x_shape({5, 2}); in TEST_F() local
128 TensorShape x_shape({5, 2, 5, 2}); in TEST_F() local
136 TensorShape x_shape({5, 2}); in TEST_F() local
[all …]
|
| D | nn_grad_test.cc | 54 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
211 TensorShape x_shape({5, 2}); in TEST_F() local
237 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
250 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
263 TensorShape x_shape({1, 3, 3, 3, 1}); in TEST_F() local
276 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
287 TensorShape x_shape({1, 3, 3, 3, 1}); in TEST_F() local
298 TensorShape x_shape({1, 1, 2, 1}); in TEST_F() local
319 TensorShape x_shape({1, 3, 7, 1}); in TEST_F() local
331 TensorShape x_shape({1, 3, 7, 1}); in TEST_F() local
|
| D | image_grad_test.cc | 81 TensorShape x_shape({1, 2, 2, 1}); in TestResizedShapeForType() local
114 TensorShape x_shape({1, 4, 6, 1}); in TestResizeToSmallerAndAlign() local
129 TensorShape x_shape({1, 2, 3, 1}); in TestResizeToLargerAndAlign() local
211 void TestScaleAndTranslate(const TensorShape x_shape, const int out_height, in TestScaleAndTranslate()
326 TensorShape x_shape({1, 4, 2, 1}); in TestCropAndResize() local
|
| D | math_grad_test.cc | 580 TensorShape x_shape = shapes[0]; in TestMatMulGrad() local
609 TensorShape x_shape; in RandMatMulShapes() local
735 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
744 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
753 TensorShape x_shape({2, 3}); in TEST_F() local
764 TensorShape x_shape({2, 3}); in TEST_F() local
783 TensorShape x_shape({1}); in TEST_F() local
793 TensorShape x_shape({1}); in TEST_F() local
839 TensorShape x_shape({3, 2, 5}); in TEST_F() local
848 TensorShape x_shape({3, 2, 5}); in TEST_F() local
[all …]
|
| D | image_grad.cc | 38 auto x_shape = Slice(scope, Shape(scope, op.input(0)), {1}, {2}); in ResizeNearestNeighborGradHelper() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker.py | 56 def _compute_theoretical_jacobian(x, x_shape, x_data, dy, dy_shape, dx, argument
134 def _compute_numeric_jacobian(x, x_shape, x_data, y, y_shape, delta, argument
211 x_shape, argument
245 x_shape, argument
271 x_shape, argument
343 x_shape, argument
|
| D | nn_fused_batchnorm_test.py | 57 x_shape, argument
109 x_shape, argument
144 def _compute_gradient_error_float16(self, x, x32, x_shape, y, y32, y_shape): argument
182 x_shape, argument
242 x_shape, argument
|
| D | image_ops_test.py | 1522 x_shape, argument
1541 x_shape, argument
1847 x_shape, argument
1866 x_shape, argument
2654 x_shape, argument
2671 x_shape, argument
3171 def _assertResizeEqual(self, x, x_shape, y, y_shape, argument
3184 def _assertResizeCheckShape(self, x, x_shape, target_shape, argument
3271 x_shape, argument
3287 x_shape, argument
[all …]
|
| D | nn_batchnorm_test.py | 303 def _testBatchNormArbitraryShapes(self, x_shape, param_shape, atol=0.0001, argument
373 def _testSuffStats(self, x_shape, axes, shift, keep_dims, has_shape): argument
|
| D | nn_test.py | 133 def testGradient(self, x_shape): argument
220 def testGradient(self, x_shape): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_op.cc | 37 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
75 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker_test.cc | 126 TensorShape x_shape({4, 3}); in TEST() local
142 TensorShape x_shape({5, 2}); in TEST() local
157 TensorShape x_shape({1, 2, 3}); in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_mul_op_test.cc | 38 void TestMul(const std::vector<int64>& x_shape, in TestMul()
87 void TestMulShape(const std::vector<int64>& x_shape, in TestMulShape()
137 void TimeMul(const std::vector<int64>& x_shape, in TimeMul()
|
| D | quantized_add_op_test.cc | 38 void TestAdd(const std::vector<int64>& x_shape, in TestAdd()
87 void TestAddShape(const std::vector<int64>& x_shape, in TestAddShape()
137 void TimeAdd(const std::vector<int64>& x_shape, in TimeAdd()
|
| D | betainc_op.cc | 49 const TensorShape& x_shape = x.shape(); in Compute() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | atrous_convolution_test.py | 254 def _test_gradient(self, x_shape, f_shape, dilation_rate, padding): argument
|
| D | sparse_tensor_dense_matmul_op_test.py | 283 def _sparse_tensor_dense_vs_dense_matmul_benchmark_sparse(x_ind, x_val, x_shape, argument
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | data_feeder.py | 48 def _get_in_out_shape(x_shape, y_shape, n_classes, batch_size=None): argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 248 std::vector<int64> x_shape({10, 100}); in TestQuantizeMul() local
283 std::vector<int64> x_shape({10, 100}); in TestQuantizeAdd() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 656 Tensor x_shape = test::AsTensor<int32>({2, 3, 4}, {3}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 191 const auto& x_shape = ShapeUtil::MakeShape(S32, {2, 4, 6}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1603 TensorShapeProto x_shape; in PredictAvgPoolGrad() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 219 def bar(x_shape): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 1547 auto x_shape = ops::Shape(s.WithOpName("grad0/Func/_1/x_shape"), grad0_z); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2695 const TensorShapeProto& x_shape = in SimplifyArithmeticOperations() local
|