Searched refs:replicated_arguments (Results 1 – 1 of 1) sorted by relevance
236 std::vector<std::vector<const ShapedBuffer*>> replicated_arguments; in ResolveAndValidateArguments() local237 replicated_arguments.resize(options_.number_of_replicas()); in ResolveAndValidateArguments()249 replicated_arguments[replica].push_back(shaped_buffer); in ResolveAndValidateArguments()252 return replicated_arguments; in ResolveAndValidateArguments()513 std::vector<absl::Span<const ShapedBuffer* const>> replicated_arguments; in ExecuteAndRegisterResult() local515 replicated_arguments.push_back(arg); in ExecuteAndRegisterResult()519 run_options, replicated_arguments)); in ExecuteAndRegisterResult()561 std::vector<std::vector<const ShapedBuffer*>> replicated_arguments, in GetArguments()563 return replicated_arguments; in GetArguments()604 TF_ASSIGN_OR_RETURN(auto replicated_arguments, in ExecuteGraphParallel()[all …]