Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 32 for WithOpName (1.04 sec)

  1. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

      Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_INT32, 0);
      Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_FLOAT, 1);
      Output arg2 = ops::_Arg(s.WithOpName("arg2"), DT_INT32, 2);
      Output add = ops::Add(s.WithOpName("add"), arg0, arg0);
      auto ret0 = ops::_Retval(s.WithOpName("ret0"), add, 0);
      auto ret1 = ops::_Retval(s.WithOpName("ret1"), arg1, 1);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/partially_decluster_pass_test.cc

      Output a = ops::Placeholder(root.WithOpName("a"), DT_FLOAT);
      Output b = ops::Shape(in_cluster_and.WithOpName("b"), a);
      Output c = ops::Rank(in_cluster_and.WithOpName("c"), b);
      Output d = ops::Size(in_cluster_and.WithOpName("d"), c);
      (void)ops::Shape(in_cluster_and.WithOpName("e"), d);
    
      auto graph = std::make_unique<Graph>(OpRegistry::Global());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 10 12:32:39 UTC 2022
    - 23K bytes
    - Viewed (0)
  3. tensorflow/cc/training/queue_runner_test.cc

      auto q0 = FIFOQueue(root.WithOpName(kQueueName), {DataType::DT_FLOAT});
      Output rnd = RandomNormal(root.WithOpName("rnd"), {1, 1}, DataType::DT_FLOAT);
      Output square = Square(root.WithOpName(kSquareOpName), rnd);
      auto enqueue0 = QueueEnqueue(root.WithOpName(kEnqueueOp0), q0, {square});
      auto close0 = QueueClose(root.WithOpName(kCloseOp0), q0);
      auto cancel0 = QueueClose(root.WithOpName(kCancelOp0), q0,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Sep 21 06:27:51 UTC 2019
    - 14.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/encapsulate_util_test.cc

      // "identity" = "add"
      tensorflow::Scope s = tensorflow::Scope::NewRootScope();
      Output const_0 = ops::Const(s.WithOpName("const_0"), 1, {2});
      Output const_1 = ops::Const(s.WithOpName("const_1"), 2, {2});
      Output add = ops::Add(s.WithOpName("add"), const_0, const_1);
      Output identity = ops::Identity(s.WithOpName("identity"), add);
      Graph g(OpRegistry::Global());
      TF_CHECK_OK(s.ToGraph(&g));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 05 22:06:22 UTC 2019
    - 2.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

        }
    
        DCHECK_EQ(slice_inputs.size_as_vector[i], -1);
    
        Output begin_i = ops::Slice(
            host_scope.WithOpName("begin_", i), slice_inputs.begin,
            constant_pool.Get1DHostConstant(i), constant_pool.Get1DHostConstant(1));
    
        Output input_shape_i = ops::Slice(
            host_scope.WithOpName("input_shape_", i), input_shape,
            constant_pool.Get1DHostConstant(i), constant_pool.Get1DHostConstant(1));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/resource_operation_safety_analysis_test.cc

      Output init_value = ops::Placeholder(root.WithOpName("init"), DT_FLOAT);
      Output loop_cond = ops::Placeholder(root.WithOpName("init"), DT_BOOL);
      Output enter_value =
          ops::internal::Enter(root.WithOpName("enter"), init_value, "fr");
      ops::Merge iv(root.WithOpName("iv"), {enter_value, enter_value});
      ops::Switch latch(root.WithOpName("latch"), iv.output, loop_cond);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 28 16:53:59 UTC 2020
    - 18.7K bytes
    - Viewed (0)
  7. tensorflow/cc/ops/const_op_test.cc

      auto x = ops::Const(root.WithOpName("x"), 1);
      EXPECT_EQ(x.node()->name(), "x");
      auto x_1 = ops::Const(root.WithOpName("x"), 1);
      EXPECT_EQ(x_1.node()->name(), "x_1");
    
      Scope child = root.NewSubScope("c");
      auto c_y = ops::Const(child.WithOpName("y"), 1);
      EXPECT_EQ(c_y.node()->name(), "c/y");
      auto c_y_1 = ops::Const(child.WithOpName("y"), 1);
      EXPECT_EQ(c_y_1.node()->name(), "c/y_1");
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 12 14:38:21 UTC 2019
    - 4.9K bytes
    - Viewed (0)
  8. tensorflow/cc/framework/cc_ops_test.cc

      // x = [1.0; 1.0]
      auto x = Const(root.WithOpName("x"), {{1.f}, {1.f}});
      // y = a * x
      auto y = MatMul(root.WithOpName("y"), a, x);
      // y2 = y.^2
      auto y2 = Square(root, y);
      // y2_sum = sum(y2)
      auto y2_sum = Sum(root, y2, 0);
      // y_norm = sqrt(y2_sum)
      auto y_norm = Sqrt(root, y2_sum);
      // y_normalized = y ./ y_norm
      auto y_normalized = Div(root.WithOpName("y_normalized"), y, y_norm);
      Tensor out;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 15 15:13:38 UTC 2023
    - 8.7K bytes
    - Viewed (0)
  9. tensorflow/cc/framework/scope_test.cc

      EXPECT_EQ(c_c.control_deps().size(), 3);
    }
    
    TEST(ScopeTest, CreateOutput) {
      Scope root = Scope::NewRootScope();
      Output a = ops::Placeholder(root.WithOpName("a"), DT_FLOAT);
      Output add;
      ASSERT_TRUE(
          CreateOutputWithScope("Add", {a, a}, root.WithOpName("add"), &add).ok());
      EXPECT_EQ(add.node()->name(), "add");
      EXPECT_EQ(add.node()->type_string(), "Add");
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 08:17:37 UTC 2019
    - 5.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/pjrt_compile_util_test.cc

      std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));
      Scope scope = Scope::NewRootScope().ExitOnError();
      auto a = ops::_Arg(scope.WithOpName("A"), DT_INT32, 0);
      auto b = ops::_Arg(scope.WithOpName("B"), DT_INT32, 1);
      auto c = ops::Add(scope.WithOpName("C"), a, b);
      auto d = ops::_Retval(scope.WithOpName("D"), c, 0);
      TF_RETURN_IF_ERROR(scope.ToGraph(graph.get()));
      return graph;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 21 23:21:57 UTC 2023
    - 4.7K bytes
    - Viewed (0)
Back to top