Home
last modified time | relevance | path

Searched defs:max_arg_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/opdefs/
H A Dtf_jitrt_ops.cc89 int64_t max_arg_size = 1; in GetMaxArgSize() local
110 int64_t max_arg_size = GetMaxArgSize(kernel_fn); in cost() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dtf_to_tfrt.cc2014 int64_t max_arg_size, in CreateCompiledModule()
2129 tf_device::ClusterOp cluster, int64_t max_arg_size, in OutlineClusterOp()
2184 int64_t max_arg_size = get_max_arg_size(parent_func); in runOnOperation() local