Search Options

Results per page
Sort
Preferred Languages
Advance

Results 111 - 120 of 126 for attrName (0.19 sec)

  1. tensorflow/c/kernels_experimental.cc

                                                    const char* attr_name,
                                                    int64_t* dims, size_t num_dims,
                                                    TF_Status* status) {
      ::tensorflow::TensorShape shape;
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx);
      ::tensorflow::Status s = cc_ctx->GetAttr(attr_name, &shape);
      ::tensorflow::Set_TF_Status_from_Status(status, s);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:12:29 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

              kEmbeddingPipelining, "_xla_outside_compilation", "device"};
          for (const auto& attr_name : attr_names) {
            if (!global_iter_id_op->hasAttr(attr_name)) continue;
            offset->setAttr(attr_name, global_iter_id_op->getAttr(attr_name));
            new_global_iter_id->setAttr(attr_name,
                                        global_iter_id_op->getAttr(attr_name));
          }
          // Make the next function to get inlined use a different offset.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/kernels/xla_ops.cc

      return *func;
    }
    
    std::vector<int> VectorAttr(OpKernelConstruction* ctx,
                                absl::string_view attr_name) {
      std::vector<int> vec;
      OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(), ctx->GetAttr(attr_name, &vec));
      return vec;
    }
    
    bool MustCompileAttr(OpKernelConstruction* ctx) {
      bool must_compile;
      OP_REQUIRES_OK_RETURN(ctx, false,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

        # Convolution ouside the while op is quantized.
        self.assertTrue(
            self._contains_op(
                output_graphdef,
                op_name='XlaConvV2',
                attr_name='RhsT',
                attr_val=attr_value_pb2.AttrValue(type=types_pb2.DT_INT8),
            )
        )
        # TODO: b/294783597 - [Converter][TF-Quantizer] Support quantization for the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/mark_for_compilation_pass.cc

      return std::nullopt;
    }
    
    // Returns true iff the attribute `attr_name` is attached to either the node or
    // to it's callee.
    static bool GetNodeOrFuncAttr(Node* node, FunctionLibraryDefinition* flib_def,
                                  const char* attr_name) {
      bool out = false;
      bool attr_value;
      if (TryGetNodeAttr(node->attrs(), attr_name, &attr_value)) {
        out |= attr_value;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  6. tensorflow/c/ops.h

    // TF_DataType, *status is populated with an error.
    TF_CAPI_EXPORT extern void TF_ShapeInferenceContext_GetAttrType(
        TF_ShapeInferenceContext* ctx, const char* attr_name, TF_DataType* val,
        TF_Status* status);
    
    // Returns the rank of the shape represented by the given handle.
    TF_CAPI_EXPORT extern int64_t TF_ShapeInferenceContextRank(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 16.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/build_xla_ops_pass.cc

                          return false;
                        }
    
                        // Only compile nodes that are marked for compilation by the
                        // compilation-marking pass (via 'attr_name').
                        return IsXlaCompiledKernel(*n);
                      });
    
      bool lazy_compilation_enabled =
          enable_lazy_compilation_
              ? *enable_lazy_compilation_
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  8. src/cmd/vendor/golang.org/x/tools/internal/stdlib/manifest.go

    		{"AttrLowerBound", Const, 0},
    		{"AttrLowpc", Const, 0},
    		{"AttrMacroInfo", Const, 0},
    		{"AttrMacros", Const, 14},
    		{"AttrMainSubprogram", Const, 14},
    		{"AttrMutable", Const, 14},
    		{"AttrName", Const, 0},
    		{"AttrNamelistItem", Const, 0},
    		{"AttrNoreturn", Const, 14},
    		{"AttrObjectPointer", Const, 14},
    		{"AttrOrdering", Const, 0},
    		{"AttrPictureString", Const, 14},
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue Apr 02 02:20:05 UTC 2024
    - 534.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      auto check = [&](mlir::Value val, std::string attr_name) -> LogicalResult {
        ElementsAttr attr;
        if (matchPattern(val, m_Constant(&attr))) {
          if (attr.getShapedType().getRank() != 1) {
            return op.emitOpError()
                   << "expects the rank of " << attr_name << "to be 1, got "
                   << attr.getShapedType().getRank();
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_experimental.h

    //
    // Analogous to TF_SetAttrValueProto for building graph operations.
    TF_CAPI_EXPORT extern void TFE_OpSetAttrValueProto(const TFE_Op* op,
                                                       const char* attr_name,
                                                       const void* proto,
                                                       size_t proto_len,
                                                       TF_Status* status);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 39.5K bytes
    - Viewed (0)
Back to top