HomeSort by relevance Sort by last modified time
    Searched refs:WithAttr (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/jit/
mark_for_compilation_pass_test.cc 104 .WithAttr("dtype", DT_FLOAT)
105 .WithAttr("value", Tensor()));
125 .WithAttr("dtype", DT_FLOAT)
126 .WithAttr("value", Tensor()));
148 .WithAttr("dtype", DT_COMPLEX128)
149 .WithAttr("value", Tensor(DT_COMPLEX128, TensorShape())));
169 .WithAttr("dtype", DT_INT32)
170 .WithAttr("value", t));
173 .WithAttr("dtype", DT_FLOAT)
174 .WithAttr("value", t))
    [all...]
encapsulate_subgraphs_pass_test.cc 340 return opts.WithAttr("value", value)
341 .WithAttr("dtype", DT_FLOAT)
350 return opts.WithAttr("Toutputs", dtypes)
351 .WithAttr("key", key)
365 return opts.WithAttr("key", key)
366 .WithAttr("Tinputs", dtypes)
395 opts.WithAttr("T", type).WithAttr("index", index));
468 Node* c = Unary(a, b1.opts().WithName("C").WithAttr("_encapsulate", "F1"));
469 Node* d = Binary(b, c, b1.opts().WithName("c").WithControlInput(c).WithAttr(
    [all...]
  /external/tensorflow/tensorflow/core/graph/
graph_def_builder.h 86 Options WithAttr(StringPiece attr_name, T&& value) const {
91 Options WithAttr(StringPiece attr_name,
93 return WithAttr<std::initializer_list<T>>(attr_name, std::move(value));
  /external/tensorflow/tensorflow/core/framework/
dataset.cc 90 opts->WithAttr("output_shapes", dataset->output_shapes())));
94 opts->WithAttr("output_types", dataset->output_dtypes())));
98 new GraphDefBuilder::Options(opts->WithAttr(attr.first, attr.second)));
185 b_->opts().WithAttr("dtype", val.dtype()).WithAttr("value", val));
  /external/tensorflow/tensorflow/core/common_runtime/
placer_test.cc 707 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"}));
730 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"}));
733 b.opts().WithName("foo").WithAttr(
752 b.opts().WithName("colocated_1").WithAttr("_class", {"loc:@in"}));
755 b.opts().WithName("foo").WithAttr(
782 b.opts().WithName("assign1").WithAttr("_class", {"loc:@var1"}));
785 b.opts().WithName("assign2").WithAttr("_class", {"loc:@var2"}));
    [all...]

Completed in 540 milliseconds