Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc299 std::map<int, Tensor> constant_args; in CompileToLocalExecutable() local
301 constant_args.insert({i, ctx->input(i)}); in CompileToLocalExecutable()
316 constant_args, *variables, ctx, &args)); in CompileToLocalExecutable()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc444 const std::map<int, Tensor>& constant_args, in BuildXlaCompilerArguments() argument
451 if (constant_args.count(input_num) > 0) { in BuildXlaCompilerArguments()
453 const Tensor& input = constant_args.at(input_num); in BuildXlaCompilerArguments()
Dxla_launch_util.h152 const std::map<int, Tensor>& constant_args,