Searched defs:other_arguments (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
H A Dfilter_dataset_op.cc40 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
41 std::vector<Tensor> other_arguments; variable
42 other_arguments.reserve(inputs.size());
44 other_arguments.push_back(t);
49 func_, std::move(other_arguments), &captured_func));
95 std::vector<Node*> other_arguments; variable
96 other_arguments.reserve(captured_func_->captured_inputs().size());
100 other_arguments.emplace_back(node);
109 this, {{0, input_graph_node}}, {{1, other_arguments}},
H A Dmap_dataset_op.cc42 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
43 std::vector<Tensor> other_arguments; variable
44 other_arguments.reserve(inputs.size());
46 other_arguments.push_back(t);
51 func_, std::move(other_arguments), &captured_func));
100 std::vector<Node*> other_arguments; variable
101 other_arguments.reserve(captured_func_->captured_inputs().size());
105 other_arguments.emplace_back(node);
115 {std::make_pair(1, other_arguments)}, // Tensor list inputs.
H A Dflat_map_dataset_op.cc43 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
44 std::vector<Tensor> other_arguments; variable
45 other_arguments.reserve(inputs.size());
47 other_arguments.push_back(t);
52 func_, std::move(other_arguments), &captured_func));
102 std::vector<Node*> other_arguments; variable
103 other_arguments.reserve(captured_func_->captured_inputs().size());
107 other_arguments.emplace_back(node);
117 {std::make_pair(1, other_arguments)}, // Tensor list inputs.
H A Dinterleave_dataset_op.cc44 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
45 std::vector<Tensor> other_arguments; variable
46 other_arguments.reserve(inputs.size());
48 other_arguments.push_back(t);
71 func_, std::move(other_arguments), &captured_func));
126 std::vector<Node*> other_arguments; variable
127 other_arguments.reserve(captured_func_->captured_inputs().size());
131 other_arguments.emplace_back(node);
142 {{1, other_arguments}},
H A Dscan_dataset_op.cc55 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
56 std::vector<Tensor> other_arguments; variable
57 other_arguments.reserve(inputs.size());
59 other_arguments.push_back(t);
64 func_, std::move(other_arguments), &captured_func));
121 std::vector<Node*> other_arguments; variable
122 other_arguments.reserve(captured_func_->captured_inputs().size());
128 other_arguments.emplace_back(node);
139 {{1, initial_state_nodes}, {2, other_arguments}},
H A Dmap_and_batch_dataset_op.cc49 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
50 std::vector<Tensor> other_arguments; variable
51 other_arguments.reserve(inputs.size());
53 other_arguments.push_back(t);
71 func_, std::move(other_arguments), &captured_func));
H A Dparallel_interleave_dataset_op.cc46 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
47 std::vector<Tensor> other_arguments; variable
48 other_arguments.reserve(inputs.size());
50 other_arguments.push_back(t);
84 func_, std::move(other_arguments), &captured_func));
H A Dparallel_map_dataset_op.cc46 OP_REQUIRES_OK(ctx, ctx->input_list("other_arguments", &inputs));
47 std::vector<Tensor> other_arguments; variable
48 other_arguments.reserve(inputs.size());
50 other_arguments.push_back(t);
62 func_, std::move(other_arguments), &captured_func));
111 // Input: other_arguments
114 std::vector<Node*> other_arguments; variable
115 other_arguments.reserve(captured_func_->captured_inputs().size());
119 other_arguments.emplace_back(node);
141 {std::make_pair(1, other_arguments)}, // Tenso
[all...]

Completed in 233 milliseconds