Searched refs:x_reshape (Results 1 – 13 of 13) 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 | 84 ndims = static_cast<int>(bcast.x_reshape().size()); in BinaryOpState()
|
| D | substr_op.cc | 149 auto input = input_tensor.shaped<tstring, 1>(bcast.x_reshape()); in Compute()
202 auto input = input_tensor.shaped<tstring, 2>(bcast.x_reshape()); in Compute()
|
| D | xent_op.cc | 89 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 | cwise_ops_common.h | 187 in0.template shaped<Tin, 2>(bcast.x_reshape()), in Compute()
194 in0.template shaped<Tin, 3>(bcast.x_reshape()), in Compute()
201 in0.template shaped<Tin, 4>(bcast.x_reshape()), in Compute()
208 in0.template shaped<Tin, 5>(bcast.x_reshape()), 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/core/util/ |
| D | matmul_bcast.h | 49 x_batch_size_ = TensorShape(batch_bcast_->x_reshape()).num_elements(); in MatMulBCast()
57 ComputeBatchIndices(output_batch_size_, batch_bcast_->x_reshape(), in MatMulBCast()
|
| D | bcast_test.cc | 33 strings::StrAppend(&ret, "[", absl::StrJoin(b.x_reshape(), ","), "]"); in BCast()
|
| D | bcast.h | 374 const Vec& x_reshape() const { return reshape_[0]; } in x_reshape() function
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | qr_op_test.py | 102 x_reshape = np.reshape(x_np, (-1, x_np.shape[-2], x_np.shape[-1]))
106 x_reshape[i, :, :], mode="complete")
109 x_reshape[i, :, :], mode="reduced")
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | qr_op_test.py | 156 x_reshape = np.reshape(x_np, (-1, x_np.shape[-2], x_np.shape[-1]))
160 x_reshape[i, :, :], mode="complete")
163 x_reshape[i, :, :], mode="reduced")
|