| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 36 const TensorShape& y_shape) { in RunTest()
62 TensorShape y_shape({2, 1, 2, 3}); in TEST_F() local
72 TensorShape y_shape({1, 2, 2, 3}); in TEST_F() local
107 TensorShape y_shape = TensorShape({5, 1}); in TEST_F() local
115 TensorShape y_shape({5, 2, 5, 2}); in TEST_F() local
123 TensorShape y_shape({5, 2}); in TEST_F() local
131 TensorShape y_shape({5, 2, 2}); in TEST_F() local
148 TensorShape y_shape({2}); in TEST_F() local
171 TensorShape y_shape({2, 5}); in TEST_F() local
179 TensorShape y_shape({1, 5, 2}); in TEST_F() local
[all …]
|
| D | nn_grad_test.cc | 47 const TensorShape& y_shape) { in RunTest()
55 const TensorShape& y_shape) { in RunTest()
156 TensorShape y_shape({1}); in TEST_F() local
182 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local
195 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local
|
| D | math_grad_test.cc | 583 TensorShape y_shape = shapes[1]; in TestMatMulGrad() local
621 TensorShape y_shape; in RandMatMulShapes() local
725 const TensorShape& y_shape) { in RunTest()
741 TensorShape y_shape({2, 5}); in TEST_F() local
750 TensorShape y_shape({2, 5}); in TEST_F() local
759 TensorShape y_shape({2}); in TEST_F() local
770 TensorShape y_shape({2}); in TEST_F() local
790 TensorShape y_shape({1}); in TEST_F() local
799 TensorShape y_shape({1}); in TEST_F() local
903 TensorShape y_shape({2, 1, 2}); in TEST_F() local
|
| D | data_flow_grad_test.cc | 67 TensorShape y_shape({3, 2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker.py | 134 def _compute_numeric_jacobian(x, x_shape, x_data, y, y_shape, delta, argument
194 def _compute_dx_and_dy(x, y, y_shape): argument
213 y_shape, argument
246 y_shape, argument
272 y_shape, argument
334 y_shape, argument
|
| D | nn_fused_batchnorm_test.py | 143 def _compute_gradient_error_float16(self, x, x32, x_shape, y, y32, y_shape): argument
|
| D | image_ops_test.py | 1418 y_shape, argument
1678 y_shape, argument
2474 y_shape, argument
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker_test.cc | 111 TensorShape y_shape({3, 2}); in TEST() local
131 TensorShape y_shape = TensorShape({5, 1}); in TEST() local
146 TensorShape y_shape({2, 1, 2, 3}); in TEST() local
|
| D | gradient_checker.cc | 114 for (const auto& y_shape : y_shapes) { in ComputeTheoreticalJacobianTranspose() local
394 const TensorShape& y_shape, JAC_T* max_error) { in ComputeGradientError()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_mul_op_test.cc | 40 float x_max_value, const std::vector<int64>& y_shape, in TestMul()
88 const std::vector<int64>& y_shape) { in TestMulShape()
138 const std::vector<int64>& y_shape, int64 iterations) { in TimeMul()
|
| D | quantized_add_op_test.cc | 40 float x_max_value, const std::vector<int64>& y_shape, in TestAdd()
88 const std::vector<int64>& y_shape) { in TestAddShape()
138 const std::vector<int64>& y_shape, int64 iterations) { in TimeAdd()
|
| /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 | 255 std::vector<int64> y_shape({100}); in TestQuantizeMul() local
290 std::vector<int64> y_shape({100}); in TestQuantizeAdd() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 262 PartialTensorShape y_shape = GetInputShape(input_y, node_map); in ShapesEqual() local
|
| D | constant_folding.cc | 1491 const TensorShapeProto& y_shape = in SimplifyGraph() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 1204 auto y_shape = ops::DynamicStitch( in TEST_F() local
|