| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_dense_binary_op_shared.cc | 88 const auto rhs_dims = BCast::FromShape(dense_t->shape()); variable
89 BCast b(lhs_dims, rhs_dims, false); // false for keeping the same num dims.
100 OP_REQUIRES(ctx, VecGreaterEq(lhs_dims, rhs_dims) && b.IsValid(),
105 str_util::Join(rhs_dims, ","), "]"));
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorConcatenation.h | 136 const Dimensions& rhs_dims = m_rightImpl.dimensions(); local
141 eigen_assert(lhs_dims[i] == rhs_dims[i]);
145 eigen_assert(rhs_dims[i] > 0);
146 m_dimensions[i] = lhs_dims[i] + rhs_dims[i];
149 eigen_assert(lhs_dims[i] == rhs_dims[i]);
161 m_rightStrides[j] = m_rightStrides[j-1] * rhs_dims[j-1];
171 m_rightStrides[j] = m_rightStrides[j+1] * rhs_dims[j+1];
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| reference_util.h | 255 std::vector<int64> rhs_dims = {rhs.n1(), rhs.n2(), rhs.n3()}; local
260 CHECK_EQ(lhs_dims[i], rhs_dims[i]);
262 out_dims[i] = lhs_dims[i] + rhs_dims[i];
289 std::vector<int64> rhs_dims = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()}; local
294 CHECK_EQ(lhs_dims[i], rhs_dims[i]);
296 out_dims[i] = lhs_dims[i] + rhs_dims[i];
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| xla_client.py | 73 def _convert_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims,
82 in zip(out_shape, window_strides, rhs_dims, lhs_dims)]
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_instruction.cc | [all...] |
| elemental_ir_emitter.cc | [all...] |
