Home
last modified time | relevance | path

Searched refs:WithAttr (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc362 return opts.WithAttr(kXlaHostTransferSequencerAttr, call_node_name) in Sequencer()
384 return opts.WithAttr("value", value) in KnownShapeBase()
385 .WithAttr("dtype", dtype) in KnownShapeBase()
405 return opts.WithAttr("shape", shape) in KeyPlaceholder()
406 .WithAttr("dtype", DT_STRING) in KeyPlaceholder()
407 .WithAttr("_host_compute_call_node", call_node) in KeyPlaceholder()
421 return opts.WithAttr("Toutputs", dtypes) in RecvAtHost()
422 .WithAttr("key", key) in RecvAtHost()
423 .WithAttr("device_ordinal", 0) in RecvAtHost()
424 .WithAttr("_encapsulate", cluster) in RecvAtHost()
[all …]
Dxla_fusion_optimizer_test.cc82 builder.opts().WithName("A").WithAttr("dtype", tensorflow::DT_FLOAT)); in TEST_F()
85 builder.opts().WithName("B").WithAttr("dtype", tensorflow::DT_FLOAT)); in TEST_F()
111 builder.opts().WithName("A").WithAttr("dtype", tensorflow::DT_FLOAT)); in TEST_F()
114 builder.opts().WithName("B").WithAttr("dtype", tensorflow::DT_FLOAT)); in TEST_F()
122 builder.opts().WithName("F").WithAttr(kXlaCompileAttr, true)); in TEST_F()
142 .WithAttr("dtype", DT_FLOAT) in TEST_F()
143 .WithAttr("value", Tensor())); in TEST_F()
166 .WithAttr("dtype", DT_FLOAT) in TEST_F()
167 .WithAttr("value", Tensor())); in TEST_F()
Dmark_for_compilation_pass_test.cc121 .WithAttr("dtype", DT_FLOAT) in TEST()
122 .WithAttr("value", Tensor())); in TEST()
142 .WithAttr("dtype", DT_FLOAT) in TEST()
143 .WithAttr("value", Tensor())); in TEST()
165 .WithAttr("dtype", DT_STRING) in TEST()
166 .WithAttr("value", Tensor(DT_STRING, TensorShape()))); in TEST()
186 .WithAttr("dtype", DT_HALF) in TEST()
187 .WithAttr("value", t)); in TEST()
387 .WithAttr("dtype", DT_FLOAT) in TEST()
388 .WithAttr("value", Tensor()) in TEST()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dcollective_order_test.cc87 .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()
Dgraph_def_builder.h86 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/framework/
Ddataset.cc203 opts->WithAttr("output_shapes", dataset->output_shapes()))); in AddDataset()
207 opts->WithAttr("output_types", dataset->output_dtypes()))); in AddDataset()
211 new GraphDefBuilder::Options(opts->WithAttr(attr.first, attr.second))); in AddDataset()
299 b_->opts().WithAttr("dtype", val.dtype()).WithAttr("shape", val.shape())); in AddPlaceholderInternal()
306 b_->opts().WithAttr("dtype", val.dtype()).WithAttr("value", val)); in AddTensorInternal()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc379 b.opts().WithName("n1").WithAttr("_class", {"loc:@in"})); in TEST_F()
1043 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"})); in TEST_F()
1066 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"})); in TEST_F()
1069 b.opts().WithName("foo").WithAttr( in TEST_F()
1088 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"})); in TEST_P()
1091 b.opts().WithName("foo").WithAttr( in TEST_P()
1128 b.opts().WithName("assign1").WithAttr("_class", {"loc:@var1"})); in TEST_F()
1131 b.opts().WithName("assign2").WithAttr("_class", {"loc:@var2"})); in TEST_F()
1159 b.opts().WithName("assign1").WithAttr("_class", {"loc:@var1"})); in TEST_P()
1162 b.opts().WithName("assign2").WithAttr("_class", {"loc:@var2"})); in TEST_P()
[all …]