| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | enable_gradient_descent.cc | 58 NodeDef& model_node = *(output->mutable_node(index)); in OptimizeAndCollectStats() local
60 if (model_node.attr().at(kAlgorithm).i() == HILL_CLIMB) { in OptimizeAndCollectStats()
61 (*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 | 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()
|
| D | latency_all_edges_test.cc | 54 NodeDef model_node = NDef("model_node", "ModelDataset", {"prefetch_node"}, in TEST_P() local
60 buffer_size_node, prefetch_node, model_node}); in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | generator_dataset_op.cc | 131 ctx, {}, &state_, model_node())); in GetNextInternal()
141 ctx, state_, out_tensors, model_node()); in GetNextInternal()
153 ctx, state_, &ignored, model_node())); in GetNextInternal()
|
| D | map_dataset_op.cc | 162 out_tensors, model_node()); in GetNextInternal()
|
| D | filter_dataset_op.cc | 151 ctx, *out_tensors, &result, model_node())); in GetNextInternal()
|
| D | flat_map_dataset_op.cc | 251 model_node()); in BuildCurrentElementIteratorLocked()
|
| D | parallel_map_dataset_op.cc | 445 std::move(done), model_node()); in CallFunction()
454 model_node()); in CallFunction()
|
| D | interleave_dataset_op.cc | 185 ¤t_elements_[cycle_index_], model_node())); in GetNextInternal()
|
| D | parallel_interleave_dataset_op.cc | 1011 model_node()); in InitializeInputs()
1305 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 | 242 ctx, next_input_element, &key_func_output, model_node())); in GetNextInternal()
259 model_node())); in GetNextInternal()
491 ctx, std::move(args), &return_values, model_node())); in StartFlushingGroup()
|
| D | take_while_dataset_op.cc | 154 ctx, *out_tensors, &result, model_node())); in GetNextInternal()
|
| D | scan_dataset_op.cc | 204 ctx, std::move(args), &state_and_output, model_node()); in GetNextInternal()
|
| D | parse_example_dataset_op.cc | 703 int64 steps = model_node() ? model_node()->num_elements() : 0; in ParseExample()
|
| D | choose_fastest_branch_dataset_op.cc | 539 ¤t_iterator_, model_node())); in MakeCurrentIterator()
|
| D | map_and_batch_dataset_op.cc | 474 std::move(done), model_node()); in CallFunction()
|
| D | parallel_interleave_dataset_op.cc | 777 model_node()); in WorkerThread()
|
| D | snapshot_dataset_op.cc | 688 ctx, tensors, &output_tensors, model_node())); in GetShardIndex()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 352 model->AddNode(std::move(factory), prefix(), parent->model_node(), &node_); in InitializeBase()
|
| D | dataset.h | 737 std::shared_ptr<model::Node> model_node() const { return node_; } in model_node() function
|