Searched refs:dimIndex (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/objc/sources/ |
| D | TFLInterpreter.mm | 205 for (int dimIndex = 0; dimIndex < shape.count; ++dimIndex) {
206 int dimension = shape[dimIndex].intValue;
214 cDimensions[dimIndex] = dimension;
313 for (int32_t dimIndex = 0; dimIndex < rank; dimIndex++) {
314 int32_t dimension = TfLiteTensorDim(cTensor, dimIndex);
318 tensorType, (unsigned long)tensor.index, dimIndex, dimension];
324 shape[dimIndex] = @((NSUInteger)dimension);
|
| /external/angle/src/compiler/translator/tree_ops/vulkan/ |
| D | RewriteArrayOfArrayOfOpaqueUniforms.cpp | 162 for (size_t dimIndex = 0; dimIndex < data.mSubArraySizes.size(); ++dimIndex) in RewriteArrayOfArraySubscriptExpression() local
166 unsigned int subArraySize = data.mSubArraySizes[dimIndex]; in RewriteArrayOfArraySubscriptExpression()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 784 auto dimIndex = dimensions.getValue<int64_t>(i); in Verify() local
785 if (dimIndex >= resultRank) { in Verify()
789 dimIndex, resultRank)); in Verify()
794 auto resultDimSize = resultType.getDimSize(dimIndex); in Verify()
799 i, dimSize, dimIndex, resultDimSize)); in Verify()
873 auto dimIndex = bcastDimensions.getValue<int64_t>(i); in Verify() local
874 if (dimIndex >= resultRank) { in Verify()
878 dimIndex, resultRank)); in Verify()
882 auto resultDimSize = resultType.getDimSize(dimIndex); in Verify()
889 i, dimSize, dimIndex, resultDimSize)); in Verify()
|