Searched refs:x_reshape (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | betainc_op.cc | 88 int ndims = static_cast<int>(a_shaper.x_reshape().size()); in Compute()
94 auto a_value = a.shaped<T, NDIM>(a_shaper.x_reshape()); \ in Compute()
95 auto b_value = b.shaped<T, NDIM>(b_shaper.x_reshape()); \ in Compute()
96 auto x_value = x.shaped<T, NDIM>(x_shaper.x_reshape()); \ in Compute()
|
| D | broadcast_to_op.h | 58 input_tensor.template shaped<T, NDIMS>(bcast.x_reshape()), in ReshapeAndBCast()
63 input_tensor.template shaped<T, NDIMS>(bcast.x_reshape()), in ReshapeAndBCast()
|
| D | cwise_ops_common.cc | 72 ndims = static_cast<int>(bcast.x_reshape().size()); in BinaryOpState()
|
| D | substr_op.cc | 144 auto input = input_tensor.shaped<string, 1>(bcast.x_reshape()); in Compute()
207 auto input = input_tensor.shaped<string, 2>(bcast.x_reshape()); in Compute()
|
| D | cwise_ops_common.h | 126 in0.template shaped<Tin, 2>(bcast->x_reshape()), in Compute()
133 in0.template shaped<Tin, 3>(bcast->x_reshape()), in Compute()
140 in0.template shaped<Tin, 4>(bcast->x_reshape()), in Compute()
147 in0.template shaped<Tin, 5>(bcast->x_reshape()), in Compute()
|
| D | xent_op.cc | 92 logits_in.template shaped<T, 2>(bcast.x_reshape()), in Compute()
|
| D | quantized_mul_op.cc | 316 const int ndims = bcast.x_reshape().size(); in Compute()
356 << TensorShape(bcast.x_reshape()).DebugString(); in Compute()
|
| D | quantized_add_op.cc | 498 const int ndims = bcast.x_reshape().size(); in Compute()
548 << TensorShape(bcast.x_reshape()).DebugString(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | 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/python/kernel_tests/ |
| D | 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")
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 95 const Vec& x_reshape() const { return x_reshape_; } in x_reshape() function
|
| D | bcast_test.cc | 33 strings::StrAppend(&ret, "[", str_util::Join(b.x_reshape(), ","), "]"); in BCast()
|