| /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({5, 2}); in TEST_F() local
120 TensorShape x_shape({5, 2, 5, 2}); in TEST_F() local
128 TensorShape x_shape({5, 2}); in TEST_F() local
145 TensorShape x_shape({2, 2}); in TEST_F() local
[all …]
|
| D | nn_grad_test.cc | 46 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest()
155 TensorShape x_shape({5, 2}); in TEST_F() local
181 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
194 TensorShape x_shape({1, 2, 2, 1}); in TEST_F() local
207 TensorShape x_shape({1, 1, 2, 1}); in TEST_F() local
|
| D | math_grad_test.cc | 582 TensorShape x_shape = shapes[0]; in TestMatMulGrad() local
611 TensorShape x_shape; in RandMatMulShapes() local
737 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
746 TensorShape x_shape({2, 3, 5, 7}); in TEST_F() local
755 TensorShape x_shape({2, 3}); in TEST_F() local
766 TensorShape x_shape({2, 3}); in TEST_F() local
785 TensorShape x_shape({1}); in TEST_F() local
795 TensorShape x_shape({1}); in TEST_F() local
841 TensorShape x_shape({3, 2, 5}); in TEST_F() local
850 TensorShape x_shape({3, 2, 5}); in TEST_F() local
[all …]
|
| /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
210 x_shape, argument
244 x_shape, argument
270 x_shape, argument
332 x_shape, argument
|
| D | nn_fused_batchnorm_test.py | 56 x_shape, argument
108 x_shape, argument
143 def _compute_gradient_error_float16(self, x, x32, x_shape, y, y32, y_shape): argument
181 x_shape, argument
241 x_shape, argument
|
| D | image_ops_test.py | 1414 x_shape, argument
1433 x_shape, argument
1674 x_shape, argument
1693 x_shape, argument
2472 x_shape, argument
2488 x_shape, argument
|
| D | nn_batchnorm_test.py | 296 def _testBatchNormArbitraryShapes(self, x_shape, param_shape, atol=0.0001): argument
359 def _testSuffStats(self, x_shape, axes, shift, keep_dims, has_shape): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_op.cc | 36 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
77 const TensorShape x_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker_test.cc | 110 TensorShape x_shape({4, 3}); in TEST() local
126 TensorShape x_shape({5, 2}); in TEST() local
141 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 | 252 def _test_gradient(self, x_shape, f_shape, dilation_rate, padding): argument
|
| D | sparse_tensor_dense_matmul_op_test.py | 276 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 | 42 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 | 608 Tensor x_shape = test::AsTensor<int32>({2, 3, 4}, {3}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 261 PartialTensorShape x_shape = GetInputShape(input_x, node_map); in ShapesEqual() local
|
| D | constant_folding.cc | 1513 const TensorShapeProto& x_shape = in SimplifyGraph() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 1203 auto x_shape = ops::Shape(s.WithOpName("grad0/Func/_1/x_shape"), grad0_z); in TEST_F() local
|