Home
last modified time | relevance | path

Searched refs:getDimSize (Results 1 – 25 of 68) sorted by relevance

123

/external/angle/third_party/vulkan-deps/glslang/src/glslang/Include/
Darrays.h163 unsigned int getDimSize(int i) const in getDimSize() function
239 int getDimSize(int dim) const { return sizes.getDimSize(dim); } in getDimSize() function
249 assert(sizes.getDimSize(d) != UnsizedArraySize); in getCumulativeSize()
250 size *= sizes.getDimSize(d); in getCumulativeSize()
267 if (sizes.getDimSize(d) == (unsigned)UnsizedArraySize) in isInnerUnsized()
276 if (sizes.getDimSize(d) == (unsigned)UnsizedArraySize) in clearInnerUnsized()
312 if (sizes.getDimSize(d) != rhs.sizes.getDimSize(d) || in sameInnerArrayness()
/external/deqp-deps/glslang/glslang/Include/
Darrays.h163 unsigned int getDimSize(int i) const in getDimSize() function
239 int getDimSize(int dim) const { return sizes.getDimSize(dim); } in getDimSize() function
249 assert(sizes.getDimSize(d) != UnsizedArraySize); in getCumulativeSize()
250 size *= sizes.getDimSize(d); in getCumulativeSize()
267 if (sizes.getDimSize(d) == (unsigned)UnsizedArraySize) in isInnerUnsized()
276 if (sizes.getDimSize(d) == (unsigned)UnsizedArraySize) in clearInnerUnsized()
312 if (sizes.getDimSize(d) != rhs.sizes.getDimSize(d) || in sameInnerArrayness()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils_test.cc137 transpose_op->getOperand(0).getType().cast<RankedTensorType>().getDimSize( in TEST_F()
141 transpose_op->getOperand(0).getType().cast<RankedTensorType>().getDimSize( in TEST_F()
145 transpose_op->getResult(0).getType().cast<RankedTensorType>().getDimSize( in TEST_F()
149 transpose_op->getResult(0).getType().cast<RankedTensorType>().getDimSize( in TEST_F()
188 EXPECT_EQ(output_types[0].cast<RankedTensorType>().getDimSize(i), in TEST_F()
248 EXPECT_EQ(it->getOperand(20).getType().cast<RankedTensorType>().getDimSize(0), in TEST_F()
257 EXPECT_EQ(output_types[0].cast<RankedTensorType>().getDimSize(i), in TEST_F()
Dlstm_utils.cc100 return transpose_type.getDimSize(dim); in Transpose()
493 if (weight_type_.getDimSize(1) % num_gates_ != 0) { in Initialize()
498 n_cell_ = weight_type_.getDimSize(1) / num_gates_; in Initialize()
505 n_output_ = projection_type_.getDimSize(1); in Initialize()
507 n_input_ = weight_type_.getDimSize(0) - n_output_; in Initialize()
508 num_cols_weight_transposed_ = weight_type_.getDimSize(0); in Initialize()
509 num_cols_projection_transposed_ = projection_type_.getDimSize(0); in Initialize()
599 int dim = input_type.getDimSize(i); in CreateEqualSizeSplitVOp()
666 int batch = time_majored ? final_input_type.getDimSize(1) in ConvertKerasLSTMLayer()
667 : final_input_type.getDimSize(0); in ConvertKerasLSTMLayer()
[all …]
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorOps.cpp70 int64_t u = m.getType().cast<VectorType>().getDimSize(0); in get1DMaskFormat()
279 lhsType.getDimSize(dimPair.first) != rhsType.getDimSize(dimPair.second)) in verifyDimMap()
305 expectedResultDims.push_back(lhsType.getDimSize(i)); in verifyOutputShape()
312 expectedResultDims.push_back(rhsType.getDimSize(i)); in verifyOutputShape()
634 attr.getInt() >= op.getVectorType().getDimSize(en.index())) in verify()
1025 if (op.getSourceVectorType().getDimSize(i) % in verify()
1026 op.getResultType().getDimSize(i) != in verify()
1030 if (op.getSourceVectorType().getDimSize(i) != in verify()
1031 op.getResultType().getDimSize(i)) in verify()
1051 if (getSourceVectorType().getDimSize(i) != getResultType().getDimSize(i)) in getMultiplicity()
[all …]
DVectorTransforms.cpp103 adjustedShape.push_back(tp.getDimSize(i)); in adjustType()
126 for (int64_t d = 0, e = resType.getDimSize(0); d < e; d++) { in reshapeLoad()
153 for (int64_t d = 0, e = type.getDimSize(0); d < e; d++) { in reshapeStore()
1164 for (int64_t d = 0, dim = dstType.getDimSize(0); d < dim; ++d) in matchAndRewrite()
1174 if (srcType.getDimSize(r) != dstType.getDimSize(r)) { in matchAndRewrite()
1216 for (int64_t d = 0, dim = dstType.getDimSize(0); d < dim; ++d) in matchAndRewrite()
1220 for (int64_t d = 0, dim = dstType.getDimSize(0); d < dim; ++d) { in matchAndRewrite()
1301 for (int64_t d = 0, e = resType.getDimSize(pos); d < e; ++d) { in expandIndices()
1351 for (int64_t d = 0, e = resType.getDimSize(0); d < e; ++d) { in matchAndRewrite()
1402 int64_t trueDim = std::min(dstType.getDimSize(0), in matchAndRewrite()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc625 const int64_t operand_dim_size = operand_type.getDimSize(dim); in VerifyConcatenationOpTypes()
655 const int64_t output_concated_dim_size = output_type.getDimSize(axis); in VerifyConcatenationOpTypes()
720 op.getType().cast<RankedTensorType>().getDimSize(axis); in ConstFoldConcatenateOpDense()
781 if (attribute_size != opaque_attr.getType().cast<ShapedType>().getDimSize(0)) in Verify()
803 const int z_in = filter_type.getDimSize(1); in Verify()
821 const int z_out = filter_type.getDimSize(0); in Verify()
1046 auto outermost_dim = indices_type.getDimSize(outer_dims); in Verify()
1050 if (indices_type.getDimSize(i) != updates_type.getDimSize(i)) { in Verify()
1052 << "indices.Dims(" << i << ") == " << indices_type.getDimSize(i) in Verify()
1054 << ") == " << updates_type.getDimSize(i); in Verify()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Doptimize_mhlo.cc76 dimension_numbers.start_index_map().getType().getDimSize(0) != 1 || in matchAndRewrite()
130 for (int i = 0; i < gather_start_indices_ty.getDimSize(0); i++) { in matchAndRewrite()
159 while (slice_start_indices.size() < sliceSizesTy.getDimSize(0)) { in matchAndRewrite()
Dlegalize_to_standard.cc121 auto max_dim_size = output_type.getDimSize(dimension); in matchAndRewrite()
136 increase_stride /= output_type.getDimSize(i); in matchAndRewrite()
Dlhlo_legalize_to_gpu.cc63 auto dim_size = shaped_type.getDimSize(0); in matchAndRewrite()
80 shaped_type.getDimSize(reducing_dimension.getSExtValue()); in matchAndRewrite()
Dlegalize_gather_to_torch_index_select.cc98 if (it.value().getSExtValue() != operand_ty.getDimSize(it.index())) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc258 input_type.getDimSize(ifm_dim), filter_type.getDimSize(filter_dim), in getPaddingValuesFromPadType()
324 int64_t ifm_size = input_type.getDimSize(ifm_dim); in getTransposeConv2dPaddingValues()
325 int64_t filter_size = filter_type.getDimSize(filter_dim); in getTransposeConv2dPaddingValues()
326 int64_t ofm_size = output_type.getDimSize(ofm_dim); in getTransposeConv2dPaddingValues()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc286 int64_t input_batch = input_type.getDimSize(0); in Verify()
307 return crops_type.getDimSize(dim) == size; in Verify()
478 int64_t feature_dim = value_ty.getDimSize(feature_dim_idx); in Verify()
479 int64_t bias_len = bias_ty.getDimSize(0); in Verify()
674 if (r0_ty && r0_ty.hasStaticShape() && r0_ty.getDimSize(0) != r0.size()) in Verify()
677 if (r1_ty && r1_ty.hasStaticShape() && r1_ty.getDimSize(0) != r1.size()) in Verify()
1349 int64_t ranked_shape_dim = ranked_shape.getDimSize(0); in Verify()
1527 input_channels = ty.getDimSize(idx); in Verify()
1534 filter_channels = ty.getDimSize(idx); in Verify()
1638 input_ty.getDimSize(GetTensorBatchDimIndex(num_dims, format)); in inferConvReturnTypes()
[all …]
Dtf_ops_n_z.cc595 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType()
859 result_ranked_type.getDimSize(0) != in VerifyShapeOperandAndResult()
1380 int64_t features_batches = features_ty.getDimSize(0); in Verify()
1381 int64_t labels_batches = labels_ty.getDimSize(0); in Verify()
1440 op.value().getType().cast<RankedTensorType>().getDimSize(*dim_index); in Verify()
1460 (split_sizes_type.getDimSize(0) != ShapedType::kDynamicSize && in Verify()
1461 split_sizes_type.getDimSize(0) != op.getNumResults())) in Verify()
1470 op.value().getType().cast<RankedTensorType>().getDimSize(*dim_index); in Verify()
1602 int64_t length = operand_ty.getDimSize(0); in VerifyStridedSliceBase()
1980 sub_shape.push_back(tensor_ty.getDimSize(i)); in fold()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc281 getResult().getType().cast<RankedTensorType>(), type.getDimSize(dim)); in fold()
317 {result_ty.getDimSize(iota_dimension)}, result_ty.getElementType()); in matchAndRewrite()
339 if (result_ty.hasRank() && result_ty.getDimSize(dimension) == 1) { in fold()
404 {result_ty.getDimSize(iota_dimension_int)}, result_ty.getElementType()); in matchAndRewrite()
682 auto split_dim_size = type.getDimSize(op.split_dimension()); in Verify()
793 auto dimSize = operandType.getDimSize(i); in Verify()
794 auto resultDimSize = resultType.getDimSize(dimIndex); in Verify()
844 auto outputDimensionsSize = outputDimensionsType.getDimSize(0); in Verify()
881 auto dimSize = operandType.getDimSize(i); in Verify()
882 auto resultDimSize = resultType.getDimSize(dimIndex); in Verify()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_exporter.cc100 std::vector<ReplicaGroup> replica_groups(type.getDimSize(0)); in ConvertReplicaGroups()
102 for (int64 element_idx = 0; element_idx < type.getDimSize(1); in ConvertReplicaGroups()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc428 return filter_type.getDimSize(dim); in legalizeFilter()
461 auto multiplier = filter.getType().cast<RankedTensorType>().getDimSize(3); in createTFLOp()
864 shape_type.getDimSize(0) <= 4))) in matchAndRewrite()
1087 int64_t v_last_dim = v_type.getDimSize(v_type.getRank() - 1); in matchAndRewrite()
1152 int64_t rest_size = input_type.getDimSize(0) * in matchAndRewrite()
1153 input_type.getDimSize(1) * input_type.getDimSize(2); in matchAndRewrite()
1390 expanded_input_shape.push_back(input_type.getDimSize(r)); in matchAndRewrite()
1391 expanded_output_shape.push_back(output_type.getDimSize(r)); in matchAndRewrite()
Ddense_to_sparse.cc108 shape[0] = type.getDimSize(0); in CalculateBlockSparsity()
109 shape[1] = type.getDimSize(1); in CalculateBlockSparsity()
234 shape[i] = type.getDimSize(i); in BuildSparsityParameterAttribute()
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/estimators/
Darithmetic_count_util.h65 cols *= output_type.getDimSize(i); in GetArithmeticCountForConvAndFullyconnectedOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_array_ops_decomposition.cc91 int64_t length = buffer_type.getDimSize(0) / *count; in GetSplitElementTypeAndCount()
300 shape[0] *= buffer_type.getDimSize(0); in HandleTensorArrayConcatV3Op()
312 {buffer_type.getDimSize(0)}); in HandleTensorArrayConcatV3Op()
313 for (int64_t i = 0; i < buffer_type.getDimSize(0); ++i) { in HandleTensorArrayConcatV3Op()
314 lengths_tensor.vec<tensorflow::int64>()(i) = buffer_type.getDimSize(1); in HandleTensorArrayConcatV3Op()
372 auto result = cutil::CreateScalarConst(buffer_type.getDimSize(0), builder, in HandleTensorArraySizeV3Op()
390 buffer_type.getShape().drop_front(), buffer_type.getDimSize(0), op, in CreateAndInitializeGradVariable()
Dlegalize_hlo.cc96 conv_op.lhs().getType().cast<ShapedType>().getDimSize( in matchAndRewrite()
145 conv_op.lhs().getType().cast<ShapedType>().getDimSize(dim), in IsSamePadding()
146 conv_op.rhs().getType().cast<ShapedType>().getDimSize(i), in IsSamePadding()
400 batch_dimensions_.sizes.push_back(type.getDimSize(dim)); in DotDimensionsInfo()
405 contracting_dimensions_.sizes.push_back(type.getDimSize(dim)); in DotDimensionsInfo()
414 out_dimensions_.sizes.push_back(type.getDimSize(dim)); in DotDimensionsInfo()
/external/llvm-project/mlir/lib/Dialect/Quant/Utils/
DUniformSupport.cpp65 size_t dimSize = type.getDimSize(quantizationDim); in convert()
/external/llvm-project/mlir/lib/Analysis/
DUtils.cpp152 int64_t dimSize = memRefType.getDimSize(r); in getConstantBoundingSizeAndShape()
173 auto dimSize = memRefType.getDimSize(d); in getConstantBoundingSizeAndShape()
379 cst.addConstantUpperBound(/*pos=*/r, memRefType.getDimSize(r) - 1); in compute()
442 sizeInBytes = sizeInBytes * memRefType.getDimSize(i); in getMemRefSizeInBytes()
475 int64_t dimSize = loadOrStoreOp.getMemRefType().getDimSize(r); in boundCheckLoadOrStoreOp()
/external/llvm-project/mlir/lib/Parser/
DAttributeParser.cpp976 ? RankedTensorType::get({indicesType.getDimSize(0)}, valuesEltType) in parseSparseElementsAttr()
985 (type.getRank() == indicesType.getDimSize(1)); in parseSparseElementsAttr()
986 auto sameElementNum = indicesType.getDimSize(0) == valuesType.getDimSize(0); in parseSparseElementsAttr()
/external/llvm-project/mlir/include/mlir/Dialect/Quant/
DQuantOps.td241 if (layerStatsType.getRank() != 1 || layerStatsType.getDimSize(0) != 2) {
258 axisStatsType.getDimSize(1) != 2 ||
259 axisStatsType.getDimSize(0) != argSliceSize) {

123