OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:rhs_dims
(Results
1 - 3
of
3
) sorted by null
/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/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/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];
Completed in 300 milliseconds