| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 350 return opts.WithAttr(kXlaHostTransferSequencerAttr, call_node_name) in Sequencer()
372 return opts.WithAttr("value", value) in KnownShapeBase()
373 .WithAttr("dtype", dtype) in KnownShapeBase()
393 return opts.WithAttr("shape", shape) in KeyPlaceholder()
394 .WithAttr("dtype", DT_STRING) in KeyPlaceholder()
395 .WithAttr("_host_compute_call_node", call_node) in KeyPlaceholder()
411 return opts.WithAttr("Toutputs", dtypes) in RecvAtHost()
412 .WithAttr("key", key) in RecvAtHost()
413 .WithAttr("device_ordinal", 0) in RecvAtHost()
414 .WithAttr("_encapsulate", cluster) in RecvAtHost()
[all …]
|
| D | cluster_scoping_pass_test.cc | 71 .WithAttr("dtypes", std::move(dtypes)); in BuildStageNode()
97 .WithAttr("dtype", DT_FLOAT) in TEST()
98 .WithAttr("value", Tensor())); in TEST()
101 .WithAttr("dtype", DT_FLOAT) in TEST()
102 .WithAttr("value", Tensor())); in TEST()
105 builder.opts().WithName("unstage").WithAttr("dtypes", {DT_FLOAT})); in TEST()
141 .WithAttr("dtype", DT_FLOAT) in TEST()
142 .WithAttr("value", Tensor())); in TEST()
145 .WithAttr("dtype", DT_FLOAT) in TEST()
146 .WithAttr("value", Tensor())); in TEST()
[all …]
|
| D | mark_for_compilation_pass_test.cc | 124 .WithAttr("dtype", DT_FLOAT) in TEST()
125 .WithAttr("value", Tensor())); in TEST()
144 .WithAttr("dtype", DT_FLOAT) in TEST()
145 .WithAttr("value", Tensor())); in TEST()
166 .WithAttr("dtype", DT_STRING) in TEST()
167 .WithAttr("value", Tensor(DT_STRING, TensorShape()))); in TEST()
186 .WithAttr("dtype", DT_HALF) in TEST()
187 .WithAttr("value", t)); in TEST()
297 .WithAttr("dtype", DT_FLOAT) in TEST()
298 .WithAttr("shape", TensorShape({}))); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order_test.cc | 87 .WithAttr("T", DT_FLOAT) in CollectiveReduceNode()
88 .WithAttr("group_size", 2) in CollectiveReduceNode()
89 .WithAttr("group_key", 1) in CollectiveReduceNode()
90 .WithAttr("instance_key", instance_key) in CollectiveReduceNode()
91 .WithAttr("merge_op", "Add") in CollectiveReduceNode()
92 .WithAttr("final_op", "Id") in CollectiveReduceNode()
93 .WithAttr("subdiv_offsets", {1})); in CollectiveReduceNode()
122 builder.opts().WithName("id0").WithDevice(dev0).WithAttr("T", DT_FLOAT)); in InitGraph()
125 builder.opts().WithName("id1").WithDevice(dev1).WithAttr("T", DT_FLOAT)); in InitGraph()
175 builder.opts().WithName("id").WithDevice(dev0).WithAttr("T", DT_FLOAT)); in InitGraph2()
|
| D | graph_def_builder.h | 86 Options WithAttr(StringPiece attr_name, T&& value) const { in WithAttr() function
91 Options WithAttr(StringPiece attr_name, in WithAttr() function
93 return WithAttr<std::initializer_list<T>>(attr_name, std::move(value)); in WithAttr()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | serialization_utils.cc | 108 .WithAttr("T", DT_VARIANT) in AsGraphDef()
109 .WithAttr("index", 0)); in AsGraphDef()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 226 opts->WithAttr("output_shapes", dataset->output_shapes()))); in AddDataset()
230 opts->WithAttr("output_types", dataset->output_dtypes()))); in AddDataset()
234 new GraphDefBuilder::Options(opts->WithAttr(attr.first, attr.second))); in AddDataset()
319 b_->opts().WithAttr("dtype", val.dtype()).WithAttr("shape", val.shape())); in AddPlaceholderInternal()
326 b_->opts().WithAttr("dtype", val.dtype()).WithAttr("value", val)); in AddTensorInternal()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_test.cc | 492 b.opts().WithName("n1").WithAttr("_class", {"loc:@in"})); in TEST_F()
1222 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"})); in TEST_F()
1245 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"})); in TEST_F()
1248 b.opts().WithName("foo").WithAttr( in TEST_F()
1267 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"})); in TEST_F()
1270 b.opts().WithName("foo").WithAttr("_class", {"loc:@colocated_1"})); in TEST_F()
1288 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"})); in TEST_P()
1291 b.opts().WithName("foo").WithAttr( in TEST_P()
1331 b.opts().WithName("assign1").WithAttr("_class", {"loc:@var1"})); in TEST_F()
1334 b.opts().WithName("assign2").WithAttr("_class", {"loc:@var2"})); in TEST_F()
[all …]
|