Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtensorflow_abi_knowledge_propagation.cc128 Value &stride_val = constants[const_stride->second]; in runOnFunction() local
129 if (!stride_val) { in runOnFunction()
130 stride_val = b.create<LLVM::ConstantOp>( in runOnFunction()
134 inner_stride.replaceAllUsesWith(stride_val); in runOnFunction()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h2613 const int stride_val = function_params->stride;
2648 workspace_height_stride * k_height * stride_val +
2658 const bool no_right_block = (output_width - 1) * stride_val < 2;
2659 TFLITE_DCHECK_LE(output_width * stride_val, 4);
2664 ConcatenateInputSubBlocks(x * stride_val, s,
2775 const int stride_val = function_params->stride;
2809 workspace_height_stride * k_height * stride_val;
2821 TFLITE_DCHECK_LE(output_width * stride_val, 4);
2825 ConcatenateInputSubBlocks(x * stride_val, workspace_height_stride,
2864 const int stride_val = function_params->stride;
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc118 Value stride_val = b->create<ConstantIndexOp>(loc, stride.getSExtValue()); in MapWindowIvsToInput() local
121 Value center = b->create<MulIOp>(loc, ivs[i], stride_val); in MapWindowIvsToInput()