Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc92 llvm::function_ref<bool(int64_t)> arg_buffer_size_is_fixed) { in ModifyFunctionSignature() argument
104 arg_buffer_size_is_fixed(i)}; in ModifyFunctionSignature()
176 auto arg_buffer_size_is_fixed = [&](int64_t index) { in HandleWhileOp() local
181 find_arg_tensor_list_type, arg_buffer_size_is_fixed); in HandleWhileOp()
193 find_arg_tensor_list_type, arg_buffer_size_is_fixed); in HandleWhileOp()
237 auto arg_buffer_size_is_fixed = [&](int64_t index) { in HandleCaseOrIfOp() local
245 find_arg_buffer_type, arg_buffer_size_is_fixed); in HandleCaseOrIfOp()
490 auto arg_buffer_size_is_fixed = [&](int64_t index) { in HandlePartitionedCallOp() local
495 arg_buffer_size_is_fixed); in HandlePartitionedCallOp()