Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc46 static int64_t multiply_dims(int64_t a, int64_t b) { in multiply_dims() function
53 static int64_t multiply_dims(llvm::ArrayRef<int64_t> dims, int64_t res = 1) { in multiply_dims() function
914 a4_reshape_shape[0] = multiply_dims(batch_size, block_num_elems); in convertSpaceToBatchNDOp()
1138 a4_shape[1 + i] = multiply_dims(input_shape[i + 1], block_shape[i]); in convertBatchToSpaceNDOp()
3504 int64_t N = multiply_dims(params_type.getShape().take_front(batch_dims)); in convertGatherOp()
3505 int64_t W = multiply_dims( in convertGatherOp()
3509 int64_t C = multiply_dims( in convertGatherOp()
3511 C = multiply_dims( in convertGatherOp()