Home
last modified time | relevance | path

Searched refs:m_dim (Results 1 – 5 of 5) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorChipping.h89 : m_xpr(expr), m_offset(offset), m_dim(dim) {
95 const Index dim() const { return m_dim.actualDim(); }
123 const internal::DimensionId<DimId> m_dim;
153 : m_impl(op.expression(), device), m_dim(op.dim()), m_device(device)
156 eigen_assert(NumInputDims > m_dim.actualDim());
159 eigen_assert(op.offset() < input_dims[m_dim.actualDim()]);
163 if (i != m_dim.actualDim()) {
172 for (int i = 0; i < m_dim.actualDim(); ++i) {
177 for (int i = NumInputDims-1; i > m_dim.actualDim(); --i) {
182 m_inputStride *= input_dims[m_dim.actualDim()];
[all …]
/external/deqp/external/vulkancts/modules/vulkan/compute/
DvktComputeZeroInitializeWorkgroupMemoryTests.cpp946 m_dim(dim) in MaxWorkgroupsInstance()
951 Dim m_dim; member in vkt::compute::__anonf49e12520111::MaxWorkgroupsInstance
962 m_dim(dim) in MaxWorkgroupsTest()
969 return new MaxWorkgroupsInstance(context, m_dim); in createInstance()
973 Dim m_dim; member in vkt::compute::__anonf49e12520111::MaxWorkgroupsTest
991 switch (m_dim) { in initPrograms()
1038 deUint32 num_wgx = m_dim == DimX ? 65535 : 1; in iterate()
1039 deUint32 num_wgy = m_dim == DimY ? 65535 : 1; in iterate()
1040 deUint32 num_wgz = m_dim == DimZ ? 65535 : 1; in iterate()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1057 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local
1074 m_dim = a_matrix_shape.dim(1).size(); in CountMatMulOperations()
1077 m_dim = a_matrix_shape.dim(0).size(); in CountMatMulOperations()
1088 VLOG(1) << "M, N, K: " << m_dim << "," << n_dim << "," << k_dim; in CountMatMulOperations()
1099 ops = m_dim * n_dim * k_dim * 2; in CountMatMulOperations()
1103 mat_mul->m = m_dim; in CountMatMulOperations()
1183 TensorShapeProto_Dim m_dim; in GenerateBatchMatmulContextFromEinsum() local
1187 m_dim.set_size(1); in GenerateBatchMatmulContextFromEinsum()
1199 m_dim.set_size(m_dim.size() * a_input_shape.dim(i_idx).size()); in GenerateBatchMatmulContextFromEinsum()
1224 *(a_matrix_shape->add_dim()) = m_dim; in GenerateBatchMatmulContextFromEinsum()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc148 int64 m_dim = (left_side) ? -1 : -2; in SolveWithInvertedDiagonalBlocks() local
149 int64 m = ShapeUtil::GetDimension(b_shape, m_dim); in SolveWithInvertedDiagonalBlocks()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_patterns.td415 [(HLO_ConstOp:$m_dim (GetDimensionSizeFromEnd<"1"> $input, $num_lower)),
422 $m_dim,