/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 114 TensorShape y_shape({2, 5, 3}); in TEST_F() local 123 TensorShape y_shape({5, 2, 5, 2}); in TEST_F() local 131 TensorShape y_shape({5, 2}); in TEST_F() local 139 TensorShape y_shape({5, 2, 2}); in TEST_F() local 156 TensorShape y_shape({2}); in TEST_F() local 179 TensorShape y_shape({2, 5}); in TEST_F() local [all …]
|
D | nn_grad_test.cc | 55 const TensorShape& y_shape) { in RunTest() 63 const TensorShape& y_shape) { in RunTest() 212 TensorShape y_shape({1}); in TEST_F() local 238 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local 251 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local 264 TensorShape y_shape({1, 1, 1, 1, 1}); in TEST_F() local 277 TensorShape y_shape({1, 1, 1, 1}); in TEST_F() local 288 TensorShape y_shape({1, 1, 1, 1, 1}); in TEST_F() local 326 TensorShape y_shape({1, 2, 3, 1}); in TEST_F() local 340 TensorShape y_shape({1, 2, 3, 1}); in TEST_F() local
|
D | data_flow_grad_test.cc | 67 TensorShape y_shape({3, 2}); in TEST_F() local
|
D | math_grad_test.cc | 581 TensorShape y_shape = shapes[1]; in TestMatMulGrad() local 619 TensorShape y_shape; in RandMatMulShapes() local 723 const TensorShape& y_shape) { in RunTest() 739 TensorShape y_shape({2, 5}); in TEST_F() local 748 TensorShape y_shape({2, 5}); in TEST_F() local 757 TensorShape y_shape({2}); in TEST_F() local 768 TensorShape y_shape({2}); in TEST_F() local 788 TensorShape y_shape({1}); in TEST_F() local 797 TensorShape y_shape({1}); in TEST_F() local 931 TensorShape y_shape({2, 1, 2}); in TEST_F() local [all …]
|
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()
|
/external/tensorflow/tensorflow/python/ops/ |
D | gradient_checker.py | 134 def _compute_numeric_jacobian(x, x_shape, x_data, y, y_shape, delta, argument 195 def _compute_dx_and_dy(x, y, y_shape): argument 214 y_shape, argument 247 y_shape, argument 273 y_shape, argument 345 y_shape, argument
|
D | gradient_checker_v2.py | 126 def _compute_theoretical_jacobian(f, y_shape, y_dtype, xs, param): argument 260 y_shape, argument
|
D | image_ops_test.py | 1526 y_shape, argument 1851 y_shape, argument 2656 y_shape, argument 2673 y_shape, argument 3171 def _assertResizeEqual(self, x, x_shape, y, y_shape, argument 3185 y_shape, preserve_aspect_ratio=True, argument 3273 y_shape, argument 3371 y_shape, argument 3472 y_shape, argument
|
D | nn_fused_batchnorm_test.py | 144 def _compute_gradient_error_float16(self, x, x32, x_shape, y, y32, y_shape): argument
|
/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 | 114 for (const auto& y_shape : y_shapes) { in ComputeTheoreticalJacobianTranspose() local 401 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 | 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 | 255 std::vector<int64> y_shape({100}); in TestQuantizeMul() local 290 std::vector<int64> y_shape({100}); in TestQuantizeAdd() local
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder_test.cc | 192 const auto& y_shape = ShapeUtil::MakeShape(S32, {2, 4}); in TEST_F() local
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | function_test.cc | 1548 auto y_shape = ops::DynamicStitch( in TEST_F() local
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | constant_folding.cc | 2663 const TensorShapeProto& y_shape = in SimplifyArithmeticOperations() local
|