Searched refs:drop_remainder_node (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_and_batch_fusion_test.cc | 99 NodeDef drop_remainder_node = output.node( in TEST() local
101 EXPECT_EQ(drop_remainder_node.attr().at("value").tensor().bool_val(0), false); in TEST()
146 NodeDef *drop_remainder_node = in TEST() local
153 batch_inputs[2] = drop_remainder_node->name(); in TEST()
269 NodeDef drop_remainder_node = output.node( in TEST() local
271 EXPECT_EQ(drop_remainder_node.attr().at("value").tensor().bool_val(0), false); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | window_dataset_op.cc | 122 Node* drop_remainder_node = nullptr; in AsGraphDefInternal() local
123 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
127 window_stride_node, drop_remainder_node}, in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op.cc | 211 Node* drop_remainder_node; in AsGraphDefInternal() local
212 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
242 std::make_pair(4, drop_remainder_node)}, // Single tensor inputs. in AsGraphDefInternal()
|
| D | numa_map_and_batch_dataset_op.cc | 162 Node* drop_remainder_node; in AsGraphDefInternal() local
163 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
191 std::make_pair(4, drop_remainder_node)}, // Single tensor inputs. in AsGraphDefInternal()
|