| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| reverse_op.cc | 37 const TensorShape x_shape = ctx->InputShape(0); variable
43 OP_REQUIRES(ctx, revd_shape.num_elements() == x_shape.dims(),
47 x_shape.DebugString(), "."));
57 for (int d = 0; d < x_shape.dims(); ++d) {
75 const TensorShape x_shape = ctx->InputShape(0); variable
81 OP_REQUIRES(ctx, axes_shape.num_elements() <= x_shape.dims(),
85 x_shape.DebugString(), "."));
97 absl::InlinedVector<bool, 8> witnessed_axes(x_shape.dims(), false);
101 ctx, (-x_shape.dims() <= axes[d]) && (axes[d] < x_shape.dims())
[all...] |
| /external/tensorflow/tensorflow/cc/gradients/ |
| image_grad.cc | 36 // x_shape = shape(x)[1:3]
38 auto x_shape = Slice(scope, Shape(scope, op.input(0)), {1}, {2}); local
40 scope, grad_inputs[0], x_shape,
|
| math_grad_test.cc | 580 TensorShape x_shape = shapes[0]; local
584 Placeholder(root_, DataTypeToEnum<T>::v(), Placeholder::Shape(x_shape));
596 root_, {x, y}, {x_shape, y_shape}, {z}, {z_shape}, &max_error)));
609 TensorShape x_shape; local
612 x_shape = tx ? TensorShape({b, k, m}) : TensorShape({b, m, k});
615 x_shape = tx ? TensorShape({k, m}) : TensorShape({m, k});
617 shapes->push_back(x_shape);
735 TensorShape x_shape({2, 3, 5, 7});
736 auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape));
740 RunTest({x}, {x_shape}, {y}, {y_shape})
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| betainc_op.cc | 49 const TensorShape& x_shape = x.shape(); variable
56 if (a_shape.dims() > 0 && x_shape.dims() > 0) {
57 OP_REQUIRES(ctx, a_shape == x_shape,
60 a_shape.DebugString(), " vs. ", x_shape.DebugString()));
62 if (b_shape.dims() > 0 && x_shape.dims() > 0) {
63 OP_REQUIRES(ctx, b_shape == x_shape,
66 b_shape.DebugString(), " vs. ", x_shape.DebugString()));
71 if (x_shape.dims() > 0) merged_shape = x_shape;
76 if (a_shape == b_shape && a_shape == x_shape) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder_test.cc | 191 const auto& x_shape = ShapeUtil::MakeShape(S32, {2, 4, 6}); local
193 auto x = Parameter(&b, 0, x_shape, "x");
198 EXPECT_TRUE(ShapeUtil::Equal(add_shape, x_shape));
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| array_grad_test.cc | 656 Tensor x_shape = test::AsTensor<int32>({2, 3, 4}, {3}); local
678 auto ddx = StridedSliceGradGrad(x_shape, start, stop, strides, dy, dx[0],
705 auto ddx = StridedSliceGradGrad(x_shape, start, stop, strides, dy, dx[0],
733 auto ddx = StridedSliceGradGrad(x_shape, start, stop, strides, dy, dx[0],
761 auto ddx = StridedSliceGradGrad(x_shape, start, stop, strides, dy, dx[0],
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function_test.cc | 1547 auto x_shape = ops::Shape(s.WithOpName("grad0\/Func\/_1\/x_shape"), grad0_z); local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| op_level_cost_estimator.cc | 1603 TensorShapeProto x_shape; local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 2695 const TensorShapeProto& x_shape = local
[all...] |
