| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 121 std::unordered_set<string> input_names; in RewriteInputsAsPlaceholders() local
122 for (const string& input_name : context.input_names) { in RewriteInputsAsPlaceholders()
123 input_names.emplace(ParseTensorName(input_name).first); in RewriteInputsAsPlaceholders()
127 if (input_names.find(node.name()) == input_names.end()) { in RewriteInputsAsPlaceholders()
152 std::unordered_set<TensorId, TensorId::Hasher> input_names; in RemoveUnusedNodes() local
153 for (const string& input : context.input_names) { in RemoveUnusedNodes()
154 input_names.insert(ParseTensorName(input)); in RemoveUnusedNodes()
175 if (input_names.count(id) > 0) { in RemoveUnusedNodes()
185 for (const TensorId& id : input_names) { in RemoveUnusedNodes()
242 std::vector<TensorId> input_names; in FoldConstants() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | auto_shard_dataset_op_test.cc | 46 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
47 input_names->clear(); in GetInputNames()
48 input_names->emplace_back(AutoShardDatasetOp::kInputDataset); in GetInputNames()
49 input_names->emplace_back(AutoShardDatasetOp::kNumWorkers); in GetInputNames()
50 input_names->emplace_back(AutoShardDatasetOp::kIndex); in GetInputNames()
|
| D | parallel_interleave_dataset_op_test.cc | 66 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
67 input_names->emplace_back(ParallelInterleaveDatasetOp::kInputDataset); in GetInputNames()
69 input_names->emplace_back( in GetInputNames()
72 input_names->emplace_back(ParallelInterleaveDatasetOp::kCycleLength); in GetInputNames()
73 input_names->emplace_back(ParallelInterleaveDatasetOp::kBlockLength); in GetInputNames()
74 input_names->emplace_back(ParallelInterleaveDatasetOp::kSloppy); in GetInputNames()
75 input_names->emplace_back( in GetInputNames()
77 input_names->emplace_back( in GetInputNames()
|
| D | map_and_batch_dataset_op_test.cc | 58 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
59 input_names->reserve(input_dataset_params_.size() + in GetInputNames()
61 input_names->emplace_back(MapAndBatchDatasetOp::kInputDataset); in GetInputNames()
63 input_names->emplace_back( in GetInputNames()
66 input_names->emplace_back(MapAndBatchDatasetOp::kBatchSize); in GetInputNames()
67 input_names->emplace_back(MapAndBatchDatasetOp::kNumParallelCalls); in GetInputNames()
68 input_names->emplace_back(MapAndBatchDatasetOp::kDropRemainder); in GetInputNames()
|
| D | assert_next_dataset_op_test.cc | 48 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
49 input_names->reserve(input_dataset_params_.size() + 1); in GetInputNames()
50 input_names->emplace_back(AssertNextDatasetOp::kInputDataset); in GetInputNames()
51 input_names->emplace_back(AssertNextDatasetOp::kTransformations); in GetInputNames()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_op.cc | 56 static const char* input_names[3] = {"basename", "shard", "num_shards"}; in Compute() local
59 errors::InvalidArgument(input_names[i], in Compute()
79 static const char* input_names[2] = {"basename", "num_shards"}; in Compute() local
82 errors::InvalidArgument(input_names[i], in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | reduce_dataset_op_test.cc | 57 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
58 input_names->clear(); in GetInputNames()
59 input_names->emplace_back("input_dataset"); in GetInputNames()
61 input_names->emplace_back(strings::StrCat("initial_state_", i)); in GetInputNames()
64 input_names->emplace_back(strings::StrCat("other_arguments_", i)); in GetInputNames()
|
| D | interleave_dataset_op_test.cc | 57 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
58 input_names->clear(); in GetInputNames()
59 input_names->reserve(input_dataset_params_.size() + in GetInputNames()
61 input_names->emplace_back(InterleaveDatasetOp::kInputDataset); in GetInputNames()
63 input_names->emplace_back( in GetInputNames()
66 input_names->emplace_back(InterleaveDatasetOp::kCycleLength); in GetInputNames()
67 input_names->emplace_back(InterleaveDatasetOp::kBlockLength); in GetInputNames()
|
| D | shuffle_dataset_op_test.cc | 57 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
58 input_names->clear(); in GetInputNames()
59 input_names->emplace_back(ShuffleDatasetOpBase::kInputDataset); in GetInputNames()
60 input_names->emplace_back(ShuffleDatasetOpBase::kBufferSize); in GetInputNames()
61 input_names->emplace_back(ShuffleDatasetOpBase::kSeed); in GetInputNames()
62 input_names->emplace_back(ShuffleDatasetOpBase::kSeed2); in GetInputNames()
64 input_names->emplace_back(ShuffleAndRepeatDatasetOp::kCount); in GetInputNames()
|
| D | shard_dataset_op_test.cc | 44 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
45 input_names->clear(); in GetInputNames()
46 input_names->emplace_back(ShardDatasetOp::kInputDataset); in GetInputNames()
47 input_names->emplace_back(ShardDatasetOp::kNumShards); in GetInputNames()
48 input_names->emplace_back(ShardDatasetOp::kIndex); in GetInputNames()
|
| D | filter_dataset_op_test.cc | 49 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
50 input_names->clear(); in GetInputNames()
51 input_names->reserve(input_dataset_params_.size() + in GetInputNames()
53 input_names->emplace_back(FilterDatasetOp::kInputDataset); in GetInputNames()
55 input_names->emplace_back( in GetInputNames()
|
| D | skip_dataset_op_test.cc | 42 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
43 input_names->clear(); in GetInputNames()
44 input_names->emplace_back(SkipDatasetOp::kInputDataset); in GetInputNames()
45 input_names->emplace_back(SkipDatasetOp::kCount); in GetInputNames()
|
| D | window_dataset_op_test.cc | 50 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
51 input_names->clear(); in GetInputNames()
52 input_names->emplace_back(WindowDatasetOp::kInputDataset); in GetInputNames()
53 input_names->emplace_back(WindowDatasetOp::kSize); in GetInputNames()
54 input_names->emplace_back(WindowDatasetOp::kShift); in GetInputNames()
55 input_names->emplace_back(WindowDatasetOp::kStride); in GetInputNames()
56 input_names->emplace_back(WindowDatasetOp::kDropRemainder); in GetInputNames()
|
| D | prefetch_dataset_op_test.cc | 48 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
49 input_names->clear(); in GetInputNames()
50 input_names->emplace_back(PrefetchDatasetOp::kInputDataset); in GetInputNames()
51 input_names->emplace_back(PrefetchDatasetOp::kBufferSize); in GetInputNames()
|
| D | map_defun_op_test.cc | 50 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
51 input_names->clear(); in GetInputNames()
53 input_names->reserve(arguments_.size() + captured_inputs_.size()); in GetInputNames()
55 input_names->emplace_back( in GetInputNames()
59 input_names->emplace_back( in GetInputNames()
|
| D | dataset_test_base.cc | 810 std::vector<string> input_names; in MakeDatasetOpKernel() local
811 TF_RETURN_IF_ERROR(dataset_params.GetInputNames(&input_names)); in MakeDatasetOpKernel()
816 name_utils::OpName(dataset_params.dataset_type(), params), input_names, in MakeDatasetOpKernel()
906 std::vector<string>* input_names) const { in GetInputNames()
907 *input_names = {RangeDatasetOp::kStart, RangeDatasetOp::kStop, in GetInputNames()
930 std::vector<string>* input_names) const { in GetInputNames()
931 *input_names = {BatchDatasetOp::kInputDataset, BatchDatasetOp::kBatchSize, in GetInputNames()
951 Status MapDatasetParams::GetInputNames(std::vector<string>* input_names) const { in GetInputNames()
952 input_names->emplace_back(MapDatasetOp::kInputDataset); in GetInputNames()
954 input_names->emplace_back( in GetInputNames()
[all …]
|
| D | sparse_tensor_slice_dataset_op_test.cc | 44 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
45 input_names->clear(); in GetInputNames()
46 input_names->emplace_back("indices"); in GetInputNames()
47 input_names->emplace_back("values"); in GetInputNames()
48 input_names->emplace_back("dense_shape"); in GetInputNames()
|
| D | parallel_interleave_dataset_op_test.cc | 64 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
65 input_names->emplace_back(ParallelInterleaveDatasetOp::kInputDataset); in GetInputNames()
67 input_names->emplace_back( in GetInputNames()
70 input_names->emplace_back(ParallelInterleaveDatasetOp::kCycleLength); in GetInputNames()
71 input_names->emplace_back(ParallelInterleaveDatasetOp::kBlockLength); in GetInputNames()
72 input_names->emplace_back(ParallelInterleaveDatasetOp::kNumParallelCalls); in GetInputNames()
|
| D | zip_dataset_op_test.cc | 46 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
47 input_names->clear(); in GetInputNames()
49 input_names->emplace_back( in GetInputNames()
|
| D | parallel_map_dataset_op_test.cc | 58 Status GetInputNames(std::vector<string>* input_names) const override { in GetInputNames()
59 input_names->emplace_back(ParallelMapDatasetOp::kInputDataset); in GetInputNames()
61 input_names->emplace_back( in GetInputNames()
64 input_names->emplace_back(ParallelMapDatasetOp::kNumParallelCalls); in GetInputNames()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_util_v2.py | 131 def resource_input_index(tensor_name, input_names, node_defs, functions): argument
155 while tensor_name not in input_names:
168 if tensor_name in input_names:
199 return input_names.index(tensor_name)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_v1_in_v2.py | 145 input_names = []
153 input_names.extend([indices_name, values_name, dense_shape_name])
155 input_names.append(original_input_name)
175 signature_fn._arg_keywords = input_names
176 if len(input_names) == 1:
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debugger_state_impl.cc | 43 const int64 executor_step_index, const std::vector<string>& input_names, in PublishDebugMetadata() argument
47 executor_step_index, input_names, in PublishDebugMetadata()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 72 NameVector input_names; in TF_Run_wrapper_helper() local
89 input_names.push_back(key_string); in TF_Run_wrapper_helper()
111 TF_Run(session, run_options, input_names.data(), inputs_unsafe.data(), in TF_Run_wrapper_helper()
112 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
117 TF_PRun(session, handle, input_names.data(), inputs_unsafe.data(), in TF_Run_wrapper_helper()
118 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
320 const NameVector& input_names, in TF_PRunSetup_wrapper() argument
326 session, const_cast<const char**>(input_names.data()), input_names.size(), in TF_PRunSetup_wrapper()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_rewriter_transform.cc | 50 for (auto i = 0; static_cast<size_t>(i) < context.input_names.size(); ++i) { in RewriteQuantizedStrippedModelForHexagon()
51 const string& input_name = context.input_names.at(i); in RewriteQuantizedStrippedModelForHexagon()
|