| /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
122 TensorShape y_shape({2, 5, 3}); in TEST_F() local
131 TensorShape y_shape({5, 2, 5, 2}); in TEST_F() local
139 TensorShape y_shape({5, 2}); in TEST_F() local
147 TensorShape y_shape({5, 2, 2}); in TEST_F() local
164 TensorShape y_shape({2}); in TEST_F() local
187 TensorShape y_shape({2, 5}); in TEST_F() local
[all …]
|
| D | linalg_grad_test.cc | 36 const TensorShape& y_shape) { in RunTest()
60 TensorShape y_shape({3, 2}); in TEST_F() local
68 TensorShape y_shape({3, 3, 2}); in TEST_F() local
74 TensorShape y_shape({3, 3}); in TEST_F() local
84 TensorShape y_shape({3, 3}); in TEST_F() local
98 TensorShape y_shape({3, 3}); in TEST_F() local
128 TensorShape y_shape({3}); in TEST_F() local
138 TensorShape y_shape({5}); in TEST_F() local
148 TensorShape y_shape({4, 5}); in TEST_F() local
|
| D | nn_grad_test.cc | 60 const TensorShape& y_shape) { in RunTest()
68 const TensorShape& y_shape) { in RunTest()
224 TensorShape y_shape({1}); in TEST_F() local
250 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local
263 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local
276 TensorShape y_shape({1, 1, 1, 1, 1}); in TEST_F() local
289 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local
300 TensorShape y_shape({1, 1, 1, 1, 1}); in TEST_F() local
338 TensorShape y_shape({1, 2, 3, 1}); in TEST_F() local
352 TensorShape y_shape({1, 2, 3, 1}); in TEST_F() local
|
| D | math_grad_test.cc | 612 TensorShape y_shape = shapes[1]; in TestMatMulGrad() local
650 TensorShape y_shape; in RandMatMulShapes() local
754 const TensorShape& y_shape) { in RunTest()
770 TensorShape y_shape({2, 5}); in TEST_F() local
779 TensorShape y_shape({2, 5}); in TEST_F() local
788 TensorShape y_shape({2}); in TEST_F() local
799 TensorShape y_shape({2}); in TEST_F() local
819 TensorShape y_shape({1}); in TEST_F() local
828 TensorShape y_shape({1}); in TEST_F() local
971 TensorShape y_shape({2, 1, 2}); in TEST_F() local
[all …]
|
| D | manip_grad_test.cc | 32 const TensorShape& y_shape) { in RunTest()
|
| D | functional_grad_test.cc | 37 const TensorShape& y_shape) { in RunTest()
|
| D | data_flow_grad_test.cc | 67 TensorShape y_shape({3, 2}); in TEST_F() local
|
| D | image_grad_test.cc | 54 void MakeOp(const OpType op_type, const Tensor& x_data, const Input& y_shape, in MakeOp()
199 void MakeOp(const Tensor& x_data, const Input& y_shape, Input scale, in MakeOp()
|
| D | linalg_grad.cc | 433 const auto y_shape = Shape(scope, y); in EinsumGrad() local
|
| D | math_grad.cc | 1209 auto y_shape = Shape(scope, y); in SelectV2Grad() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker.py | 132 def _compute_numeric_jacobian(x, x_shape, x_data, y, y_shape, delta, argument
193 def _compute_dx_and_dy(x, y, y_shape): argument
212 y_shape, argument
245 y_shape, argument
277 y_shape, argument
354 y_shape, argument
|
| D | gradient_checker_v2.py | 127 def _compute_theoretical_jacobian(f, y_shape, y_dtype, xs, param): argument
257 def _compute_gradient(f, y_shape, y_dtype, xs, param, delta): argument
|
| D | image_ops_test.py | 1786 y_shape, argument
2115 y_shape, argument
2367 y_shape, argument
3301 y_shape, argument
3318 y_shape, argument
3847 def _assertResizeEqual(self, x, x_shape, y, y_shape, argument
3861 y_shape, preserve_aspect_ratio=True, argument
3937 y_shape, argument
4057 y_shape, argument
4178 y_shape, argument
|
| D | nn_fused_batchnorm_test.py | 185 def _compute_gradient_error_float16(self, x, x32, x_shape, y, y32, y_shape): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_add_op_test.cc | 40 float x_max_value, const std::vector<int64_t>& y_shape, in TestAdd()
88 const std::vector<int64_t>& y_shape) { in TestAddShape()
138 const std::vector<int64_t>& y_shape, int64_t iterations) { in TimeAdd()
|
| D | quantized_mul_op_test.cc | 40 float x_max_value, const std::vector<int64_t>& y_shape, in TestMul()
88 const std::vector<int64_t>& y_shape) { in TestMulShape()
138 const std::vector<int64_t>& y_shape, int64_t iterations) { in TimeMul()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker_test.cc | 127 TensorShape y_shape({3, 2}); in TEST() local
147 TensorShape y_shape = TensorShape({5, 1}); in TEST() local
162 TensorShape y_shape({2, 1, 2, 3}); in TEST() local
|
| D | gradient_checker.cc | 115 for (const auto& y_shape : y_shapes) { in ComputeTheoreticalJacobianTranspose() local
408 const TensorShape& y_shape, JAC_T* max_error) { in ComputeGradientError()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | type_spec_test.py | 58 def __init__(self, x_shape, x_dtype, y_shape, y_dtype, color="red"): argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 255 std::vector<int64_t> y_shape({100}); in TestQuantizeMul() local
290 std::vector<int64_t> y_shape({100}); in TestQuantizeAdd() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 195 const auto& y_shape = ShapeUtil::MakeShape(S32, {2, 4}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 1898 auto y_shape = ops::DynamicStitch( in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 226 ArrayRef<int64_t> y_shape = y_ty.getShape(); in Verify() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2993 const TensorShapeProto& y_shape = in SimplifyArithmeticOperations() local
|