Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for IsConstant (0.2 sec)

  1. tensorflow/cc/ops/const_op_test.cc

    namespace tensorflow {
    
    namespace {
    
    template <typename T>
    void ExpectNodeEqual(const Node* n, gtl::ArraySlice<T> values,
                         TensorShape shape) {
      EXPECT_TRUE(n->IsConstant());
      Tensor tensor;
      TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor));
      DataType dtype;
      TF_EXPECT_OK(GetNodeAttr(n->attrs(), "dtype", &dtype));
      EXPECT_EQ(tensor.dtype(), dtype);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 12 14:38:21 UTC 2019
    - 4.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/aot/embedded_protocol_buffers.cc

                                             /*AddNull=*/false);
      new llvm::GlobalVariable(
          *module, protobuf_array_initializer->getType(),
          /*isConstant=*/true, llvm::GlobalValue::ExternalLinkage,
          protobuf_array_initializer, AsStringRef(*protobuf_array_symbol_name));
    }
    
    static string CreateCPPShimExpression(
        absl::string_view qualified_cpp_protobuf_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:48:41 UTC 2024
    - 5.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/clone_constants_for_better_clustering.cc

      return total_elements < kSmallTensorThreshold;
    }
    
    // We only clone small constants since we want to avoid increasing memory
    // pressure on GPUs.
    absl::StatusOr<bool> IsSmallConstant(Node* n) {
      if (!n->IsConstant()) {
        return false;
      }
    
      return IsConstantSmall(n);
    }
    
    bool IsInPlaceOp(absl::string_view op_name) {
      return op_name == "InplaceUpdate" || op_name == "InplaceAdd" ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/node_matchers_test.cc

      Output enter = ops::internal::Enter(
          root.WithOpName("enter"),
          ops::Placeholder(root.WithOpName("data"), DT_FLOAT), "frame_name",
          ops::internal::Enter::Attrs{}.IsConstant(true));
      EXPECT_THAT(enter.node(), NodeWith(Attr("is_constant", true)));
      EXPECT_EQ(Explain(enter.node(), NodeWith(Attr("is_constant", false))),
                "attribute named is_constant does not match value; expected: "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 14:43:57 UTC 2024
    - 9.1K bytes
    - Viewed (0)
Back to top