Searched refs:data_inputs (Results 1 – 6 of 6) sorted by relevance
| /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()
257 const Tensor& data = data_inputs[input_num]; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 47 std::vector<DatasetBase*> data_inputs; in MakeDataset() local
51 data_inputs.push_back(input); in MakeDataset()
54 ctx, data_inputs[0]->output_dtypes() == input->output_dtypes(), in MakeDataset()
58 DataTypeVectorString(data_inputs[0]->output_dtypes()), in MakeDataset()
62 *output = new Dataset(ctx, selector_input, std::move(data_inputs)); in MakeDataset()
69 std::vector<DatasetBase*> data_inputs) in Dataset() argument
72 data_inputs_(std::move(data_inputs)) { in Dataset()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | interleave_ops.py | 101 def __init__(self, selector_input, data_inputs): argument
103 self._data_inputs = list(data_inputs)
105 first_output_types = dataset_ops.get_legacy_output_types(data_inputs[0])
106 first_output_classes = dataset_ops.get_legacy_output_classes(data_inputs[0])
108 for data_input in data_inputs[1:]:
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 275 std::vector<std::pair<Node*, int>> data_inputs(num_inputs); in BuildXlaLaunchOps() local
283 data_inputs[i] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
291 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
353 for (int i = 0; i < data_inputs.size(); ++i) { in BuildXlaLaunchOps()
354 graph->AddEdge(data_inputs[i].first, data_inputs[i].second, xla_launch, in BuildXlaLaunchOps()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 1125 int data_inputs = 0; in DetermineCondStateMerge() local
1131 if (!e->IsControlEdge()) ++data_inputs; in DetermineCondStateMerge()
1141 if (data_inputs != 2) { in DetermineCondStateMerge()
1143 dst->name(), " only has ", data_inputs, in DetermineCondStateMerge()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 358 for (const auto& p : node_info.data_inputs) { in AddNodeInputs()
431 nodes_[node_index].data_inputs.push_back( in AddInput()
449 std::vector<std::pair<int, int>> data_inputs; member
|