Home
last modified time | relevance | path

Searched refs:lhs_dims (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_dense_binary_op_shared.cc84 const auto lhs_dims = BCast::FromShape(TensorShape(shape_vec)); in Compute() local
86 BCast b(lhs_dims, rhs_dims, false); // false for keeping the same num dims. in Compute()
97 OP_REQUIRES(ctx, VecGreaterEq(lhs_dims, rhs_dims) && b.IsValid(), in Compute()
101 str_util::Join(lhs_dims, ","), "] vs. [", in Compute()
120 const int ndims = lhs_dims.size(); in Compute()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorConcatenation.h135 const Dimensions& lhs_dims = m_leftImpl.dimensions();
140 eigen_assert(lhs_dims[i] > 0);
141 eigen_assert(lhs_dims[i] == rhs_dims[i]);
142 m_dimensions[i] = lhs_dims[i];
144 eigen_assert(lhs_dims[i] > 0); // Now i == m_axis.
146 m_dimensions[i] = lhs_dims[i] + rhs_dims[i];
148 eigen_assert(lhs_dims[i] > 0);
149 eigen_assert(lhs_dims[i] == rhs_dims[i]);
150 m_dimensions[i] = lhs_dims[i];
160 m_leftStrides[j] = m_leftStrides[j-1] * lhs_dims[j-1];
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h261 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3()}; in Concat3D() local
266 out_dims[i] = lhs_dims[i]; in Concat3D()
267 CHECK_EQ(lhs_dims[i], rhs_dims[i]); in Concat3D()
269 out_dims[i] = lhs_dims[i] + rhs_dims[i]; in Concat3D()
296 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3(), lhs.n4()}; in Concat4D() local
301 out_dims[i] = lhs_dims[i]; in Concat4D()
302 CHECK_EQ(lhs_dims[i], rhs_dims[i]); in Concat4D()
304 out_dims[i] = lhs_dims[i] + rhs_dims[i]; in Concat4D()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py250 def _convert_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims, argument
269 out_shape = np.ceil(np.true_divide(lhs_dims, window_strides)).astype(int)
272 in zip(out_shape, window_strides, rhs_dims, lhs_dims)]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc3054 std::vector<string> lhs_dims(2 + dnums.input_spatial_dimensions().size()); in ConvolutionDimensionNumbersToString() local
3055 lhs_dims[dnums.input_batch_dimension()] = 'b'; in ConvolutionDimensionNumbersToString()
3056 lhs_dims[dnums.input_feature_dimension()] = 'f'; in ConvolutionDimensionNumbersToString()
3058 lhs_dims[dnums.input_spatial_dimensions(i)] = StrCat(i); in ConvolutionDimensionNumbersToString()
3075 return StrCat(StrJoin(lhs_dims, ""), "_", StrJoin(rhs_dims, ""), "->", in ConvolutionDimensionNumbersToString()
Delemental_ir_emitter.cc2124 int64 lhs_dims = hlo->operand(0)->shape().dimensions_size(); in EmitElementalDot() local
2153 for (int64 i = 0; i < lhs_dims - 1; i++) { in EmitElementalDot()
2167 rhs_multi_index.push_back(dot_result_index[lhs_dims - 1 + i]); in EmitElementalDot()
Dalgebraic_simplifier_test.cc4320 std::vector<int64> lhs_dims = {1, 3, 5}; in TEST_P() local
4321 std::vector<int64> rhs_dims = lhs_dims; in TEST_P()
4322 std::vector<int64> output_dims = lhs_dims; in TEST_P()
4324 lhs_dims.push_back(m); in TEST_P()
4328 lhs_dims.push_back(k); in TEST_P()
4336 Shape lhs_shape = ShapeUtil::MakeShape(element_type, lhs_dims); in TEST_P()