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

  /external/tensorflow/tensorflow/cc/framework/
scope.h 141 Scope ColocateWith(const Operation& op) const;
143 Scope ColocateWith(const Output& out) const { return ColocateWith(out.op()); }
cc_ops_test.cc 166 TEST(CCOpTest, ColocateWith) {
169 auto c2 = Const(root.WithOpName("c2").ColocateWith(c1), 2);
174 auto c3 = Const(root.WithOpName("c3").ColocateWith(c2), 3);
179 auto c4 = Const(root.WithOpName("c4").ColocateWith(a), 5);
183 auto c5 = Const(root.WithOpName("c5").ColocateWith(c3).ColocateWith(c4), 6);
188 Scope with_colocate = root.ColocateWith(c3).ColocateWith(c4);
scope.cc 386 Scope Scope::ColocateWith(const Operation& op) const {
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties_test.cc 819 Output c = ops::Const(s.WithOpName("c").ColocateWith(a), 3.0f, {1});
    [all...]
  /external/tensorflow/tensorflow/c/
c_api_test.cc     [all...]

Completed in 1406 milliseconds