Searched refs:list_inputs (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 271 llvm::SmallVector<Value, 4> list_inputs; in CollectInputsAndAttributes() local
275 list_inputs.push_back(cast_op.arg()); in CollectInputsAndAttributes()
279 list_inputs); in CollectInputsAndAttributes()
280 inputs->append(list_inputs.begin(), list_inputs.end()); in CollectInputsAndAttributes()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_spec.py | 184 list_inputs = []
191 list_inputs.append(inputs[name])
192 inputs = list_inputs
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 214 const std::vector<std::pair<size_t, gtl::ArraySlice<Node*>>>& list_inputs, in AddDataset() argument
243 size_t total_size = inputs.size() + list_inputs.size(); in AddDataset()
245 auto list_inputs_iter = list_inputs.begin(); in AddDataset()
250 } else if (list_inputs_iter != list_inputs.end() && in AddDataset()
|
| D | dataset.h | 241 const std::vector<std::pair<size_t, gtl::ArraySlice<Node*>>>& list_inputs,
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parallel_interleave_dataset_op.cc | 165 std::vector<std::pair<size_t, gtl::ArraySlice<Node*>>> list_inputs; in AsGraphDefInternal() local
176 list_inputs.emplace_back(input_index++, other_arguments); in AsGraphDefInternal()
219 TF_RETURN_IF_ERROR(b->AddDataset(this, inputs, list_inputs, attrs, output)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_interleave_dataset_op.cc | 245 std::vector<std::pair<size_t, gtl::ArraySlice<Node*>>> list_inputs; in AsGraphDefInternal() local
256 list_inputs.emplace_back(input_index++, other_arguments); in AsGraphDefInternal()
303 TF_RETURN_IF_ERROR(b->AddDataset(this, inputs, list_inputs, attrs, output)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | analyzer_cli.py | 875 def list_inputs(self, args, screen_info=None): member in DebugAnalyzer
1624 analyzer.list_inputs,
|
| D | analyzer_cli_test.py | 544 analyzer.list_inputs,
|