choose_fastest_branch_dataset_op.cc | 132 OP_REQUIRES_OK(ctx, ctx->GetAttr("branches", &funcs_)); 140 ctx, funcs_.size() == other_arguments_lengths_.size(), 162 funcs_.size()); 168 for (int i = 0; i < funcs_.size(); ++i) { 177 funcs_[i], ctx, std::move(captured_args), 181 new Dataset(ctx, input, funcs_, std::move(captured_funcs), 198 funcs_(funcs), 289 b->BuildAttrValue(funcs_, &branches_attr); 290 for (const auto& func : funcs_) { 323 instantiated_captured_funcs_(dataset()->funcs_.size()) 533 std::vector<NameAttrList> funcs_; member in class:tensorflow::data::__anon45267::ChooseFastestBranchDatasetOp::Dataset 547 std::vector<NameAttrList> funcs_; member in class:tensorflow::data::__anon45267::ChooseFastestBranchDatasetOp [all...] |