| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op.cc | 77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult() argument
101 OP_REQUIRES_OK(c, c->input_list("data", data_inputs)); in CheckArgsAndAllocateResult()
102 const Tensor& data0 = (*data_inputs)[0]; in CheckArgsAndAllocateResult()
106 const Tensor& data = (*data_inputs)[input_num]; in CheckArgsAndAllocateResult()
165 OpInputList data_inputs; in Compute() local
169 this->CheckArgsAndAllocateResult(c, &indices_inputs, &data_inputs, in Compute()
203 auto data_ptr_base = data_inputs[i].template flat<T>().data(); in Compute()
237 OpInputList data_inputs; in Compute() local
240 this->CheckArgsAndAllocateResult(c, &indices_inputs, &data_inputs, in Compute()
258 const Tensor& data = data_inputs[input_num]; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 42 std::vector<DatasetBase*> data_inputs) in Dataset() argument
45 data_inputs_(std::move(data_inputs)) { in Dataset()
299 std::vector<DatasetBase*> data_inputs; in MakeDataset() local
303 data_inputs.push_back(input); in MakeDataset()
305 OP_REQUIRES(ctx, data_inputs[0]->output_dtypes() == input->output_dtypes(), in MakeDataset()
309 DataTypeVectorString(data_inputs[0]->output_dtypes()), in MakeDataset()
313 *output = new Dataset(ctx, selector_input, std::move(data_inputs)); in MakeDataset()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | interleave_ops.py | 107 def __init__(self, selector_input, data_inputs): argument
109 self._data_inputs = list(data_inputs)
111 first_output_types = dataset_ops.get_legacy_output_types(data_inputs[0])
112 first_output_classes = dataset_ops.get_legacy_output_classes(data_inputs[0])
114 for i, data_input in enumerate(data_inputs[1:]):
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 270 std::vector<std::pair<Node*, int>> data_inputs(num_inputs); in BuildXlaLaunchOps() local
278 data_inputs[i] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
286 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
348 for (int i = 0, end = data_inputs.size(); i < end; ++i) { in BuildXlaLaunchOps()
349 graph->AddEdge(data_inputs[i].first, data_inputs[i].second, xla_launch, in BuildXlaLaunchOps()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 243 std::vector<NodeBuilder::NodeOut> data_inputs; in RewriteIntoAddN() local
245 data_inputs.reserve(n->num_inputs()); in RewriteIntoAddN()
251 data_inputs.emplace_back(in_edge->src(), in_edge->src_output()); in RewriteIntoAddN()
262 .Input(data_inputs) in RewriteIntoAddN()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 2700 std::vector<std::pair<Node*, int>> data_inputs(num_inputs); in BuildTPUReplicateOps() local
2718 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildTPUReplicateOps()
2734 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildTPUReplicateOps()
2747 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildTPUReplicateOps()
2757 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildTPUReplicateOps()
2766 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildTPUReplicateOps()
2845 for (int i = 0; i < data_inputs.size(); ++i) { in BuildTPUReplicateOps()
2846 graph->AddEdge(data_inputs[i].first, data_inputs[i].second, tpu_replicate, in BuildTPUReplicateOps()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1123 std::set<const Node*> data_inputs; in AddStrictInputSemantics() local
1127 data_inputs.insert(edge->src()); in AddStrictInputSemantics()
1140 for (const Node* node : data_inputs) { in AddStrictInputSemantics()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 1172 int data_inputs = 0; in DetermineCondStateMerge() local
1178 if (!e->IsControlEdge()) ++data_inputs; in DetermineCondStateMerge()
1188 if (data_inputs != 2) { in DetermineCondStateMerge()
1190 dst->name(), " only has ", data_inputs, in DetermineCondStateMerge()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 385 for (const auto& p : node_info.data_inputs) { in AddNodeInputs()
458 nodes_[node_index].data_inputs.push_back( in AddInput()
476 std::vector<std::pair<int, int>> data_inputs; member
|