Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtensorflow_abi_knowledge_propagation.cc62 inner_stride_is_constant.insert({argument, 1}); in runOnFunction()
76 inner_stride_is_constant.insert({allocated, 1}); in runOnFunction()
123 auto const_stride = inner_stride_is_constant.find(operand); in runOnFunction()
124 if (const_stride != inner_stride_is_constant.end()) { in runOnFunction()
164 auto const_stride = inner_stride_is_constant.find(candidate); in propagateThroughUses()
165 if (const_stride != inner_stride_is_constant.end()) { in propagateThroughUses()
166 inner_stride_is_constant.insert({result, const_stride->second}); in propagateThroughUses()
188 inner_stride_is_constant.insert({result, dyn_stride.getValue()}); in propagateThroughUses()
191 inner_stride_is_constant.insert( in propagateThroughUses()
205 llvm::SmallDenseMap<Value, int64_t, 8> inner_stride_is_constant; member