| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | enable_gradient_descent.cc | 57 NodeDef& model_node = *(output->mutable_node(index)); in OptimizeAndCollectStats() local
59 if (model_node.attr().at(kAlgorithm).i() == HILL_CLIMB) { in OptimizeAndCollectStats()
60 (*model_node.mutable_attr())[kAlgorithm].set_i(GRADIENT_DESCENT); in OptimizeAndCollectStats()
|
| D | disable_intra_op_parallelism_test.cc | 185 NodeDef model_node = output.node( in TEST() local
187 EXPECT_EQ(model_node.op(), "ModelDataset"); in TEST()
188 EXPECT_EQ(model_node.input_size(), 1); in TEST()
190 graph_utils::FindGraphNodeWithName(model_node.input(0), output)); in TEST()
|
| D | use_private_thread_pool_test.cc | 181 NodeDef model_node = output.node( in TEST() local
183 EXPECT_EQ(model_node.op(), "ModelDataset"); in TEST()
184 EXPECT_EQ(model_node.input_size(), 1); in TEST()
186 graph_utils::FindGraphNodeWithName(model_node.input(0), output)); in TEST()
|
| D | enable_gradient_descent_test.cc | 70 NodeDef model_node = in TEST_P() local
72 EXPECT_EQ(model_node.attr().at("algorithm").i(), in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | generator_dataset_op.cc | 132 ctx, {}, &state_, model_node())); in GetNextInternal()
142 ctx, state_, out_tensors, model_node()); in GetNextInternal()
154 ctx, state_, &ignored, model_node())); in GetNextInternal()
|
| D | interleave_dataset_op.cc | 185 ¤t_elements_[cycle_index_], model_node())); in GetNextInternal()
391 ¤t_elements_[cycle_index_], model_node())); in MoveToNextElement()
411 ¤t_elements_[cycle_index_], model_node())); in MoveToNextElement()
|
| D | filter_dataset_op.cc | 151 ctx, *out_tensors, &result, model_node())); in GetNextInternal()
|
| D | map_dataset_op.cc | 184 out_tensors, model_node()); in GetNextInternal()
|
| D | parallel_filter_dataset_op.cc | 369 std::move(done), model_node()); in CallFunction()
378 &result->predicate_values, model_node()); in CallFunction()
|
| D | flat_map_dataset_op.cc | 299 prefix(), ¤t_element_iterator_, model_node()); in BuildCurrentElementIteratorLocked()
|
| D | parallel_map_dataset_op.cc | 485 std::move(done), model_node()); in CallFunction()
494 model_node()); in CallFunction()
|
| D | parallel_interleave_dataset_op.cc | 1109 model_node()); in InitializeInputs()
1407 model_node())); in ReadElement()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 232 ctx, next_input_element, &key_func_output, model_node())); in GetNextInternal()
248 model_node())); in GetNextInternal()
262 ctx, std::move(args), &reduce_func_output, model_node())); in GetNextInternal()
278 ctx, states_[keys_[keys_index_++]], out_tensors, model_node())); in GetNextInternal()
|
| D | group_by_window_dataset_op.cc | 243 ctx, next_input_element, &key_func_output, model_node())); in GetNextInternal()
260 model_node())); in GetNextInternal()
500 ctx->is_restoring() ? nullptr : model_node())); in StartFlushingGroup()
|
| D | take_while_dataset_op.cc | 154 ctx, *out_tensors, &result, model_node())); in GetNextInternal()
|
| D | parse_example_dataset_op.cc | 610 auto node = model_node(); in CallFunction()
714 int64_t steps = model_node() ? model_node()->num_elements() : 0; in ParseExample()
|
| D | scan_dataset_op.cc | 212 ctx, std::move(args), &state_and_output, model_node()); in GetNextInternal()
|
| D | choose_fastest_branch_dataset_op.cc | 539 ¤t_iterator_, model_node())); in MakeCurrentIterator()
|
| D | map_and_batch_dataset_op.cc | 484 std::move(done), model_node()); in CallFunction()
|
| D | parallel_interleave_dataset_op.cc | 806 model_node()); in WorkerThread()
|
| D | snapshot_dataset_op.cc | 496 ctx, tensors, &output_tensors, model_node())); in GetShardIndex()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | root_dataset.cc | 226 if (model_node() != nullptr) { in GetTraceMeMetadata()
231 model_node()->TotalMaximumBufferedBytes() / 1.0e6)))); in GetTraceMeMetadata()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.h | 826 std::shared_ptr<model::Node> model_node() const { return node_; } in model_node() function
|
| D | dataset.cc | 414 model->AddNode(std::move(factory), prefix(), parent->model_node(), &node_); in InitializeBase()
|