- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for arg_buffer_size_is_fixed (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
return it->getFirst().getType(); }; auto arg_buffer_size_is_fixed = [&](int64_t index) { return (*buffer_to_size)[while_op.getOperand(index)].fixed; }; OpBuilder builder(while_op); ModifyFunctionSignature(body, cutil::GetSizeType(builder), &body_map, find_arg_tensor_list_type, arg_buffer_size_is_fixed); if (failed(DecomposeTensorListOpsInternal(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0)