Searched refs:replicated_arguments (Results 1 – 1 of 1) sorted by relevance
234 std::vector<std::vector<const ShapedBuffer*>> replicated_arguments; in ResolveAndValidateArguments() local235 replicated_arguments.resize(options_.number_of_replicas()); in ResolveAndValidateArguments()258 replicated_arguments[replica].push_back(shaped_buffer); in ResolveAndValidateArguments()261 return replicated_arguments; in ResolveAndValidateArguments()561 std::vector<absl::Span<const ShapedBuffer* const>> replicated_arguments; in ExecuteAndRegisterResult() local563 replicated_arguments.push_back(arg); in ExecuteAndRegisterResult()567 run_options, replicated_arguments)); in ExecuteAndRegisterResult()609 std::vector<std::vector<const ShapedBuffer*>> replicated_arguments, in GetArguments()611 return replicated_arguments; in GetArguments()652 TF_ASSIGN_OR_RETURN(auto replicated_arguments, in ExecuteGraphParallel()[all …]