| /external/tensorflow/tensorflow/compiler/tests/ |
| qr_op_test.py | 85 x_reshape = np.reshape(x_np, (-1, x_np.shape[-2], x_np.shape[-1]))
89 x_reshape[i, :, :], mode="complete")
92 x_reshape[i, :, :], mode="reduced")
|
| /external/tensorflow/tensorflow/core/kernels/ |
| broadcast_to_op.h | 58 input_tensor.template shaped<T, NDIMS>(bcast.x_reshape()),
63 input_tensor.template shaped<T, NDIMS>(bcast.x_reshape()),
|
| betainc_op.cc | 88 int ndims = static_cast<int>(a_shaper.x_reshape().size());
94 auto a_value = a.shaped<T, NDIM>(a_shaper.x_reshape()); \
95 auto b_value = b.shaped<T, NDIM>(b_shaper.x_reshape()); \
96 auto x_value = x.shaped<T, NDIM>(x_shaper.x_reshape()); \
|
| cwise_ops_common.cc | 72 ndims = static_cast<int>(bcast.x_reshape().size());
|
| xent_op.cc | 92 logits_in.template shaped<T, 2>(bcast.x_reshape()),
|
| quantized_mul_op.cc | 316 const int ndims = bcast.x_reshape().size();
355 LOG(INFO) << "bcast.x_reshape()="
356 << TensorShape(bcast.x_reshape()).DebugString();
|
| substr_op.cc | 144 auto input = input_tensor.shaped<string, 1>(bcast.x_reshape());
207 auto input = input_tensor.shaped<string, 2>(bcast.x_reshape());
|
| cwise_ops_common.h | 126 in0.template shaped<Tin, 2>(bcast->x_reshape()),
133 in0.template shaped<Tin, 3>(bcast->x_reshape()),
140 in0.template shaped<Tin, 4>(bcast->x_reshape()),
147 in0.template shaped<Tin, 5>(bcast->x_reshape()),
|
| quantized_add_op.cc | 498 const int ndims = bcast.x_reshape().size();
547 LOG(INFO) << "bcast.x_reshape()="
548 << TensorShape(bcast.x_reshape()).DebugString();
|
| /external/tensorflow/tensorflow/core/util/ |
| bcast.h | 50 // output = x.reshape(b.x_reshape()).broadcast(b.x_bcast())
95 const Vec& x_reshape() const { return x_reshape_; } function in class:tensorflow::BCast
|
| bcast_test.cc | 33 strings::StrAppend(&ret, "[", str_util::Join(b.x_reshape(), ","), "]");
181 "[11,7,5,1,1,3,2,1,1][1,1,1,1,1,1,1,1,1]" // x_reshape(), x_bcast()
195 "[1,1,1,1,1,1,1,1,1][11,7,5,1,1,3,2,1,1]" // x_reshape(), x_bcast()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| qr_op_test.py | 146 x_reshape = np.reshape(x_np, (-1, x_np.shape[-2], x_np.shape[-1]))
150 x_reshape[i, :, :], mode="complete")
153 x_reshape[i, :, :], mode="reduced")
|
