- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 124 for attr_name (0.23 sec)
-
tensorflow/c/c_api_function_test.cc
func_ = TF_FunctionImportFunctionDef(buf.data(), buf.size(), s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); } void GetAttr(const char* attr_name, AttrValue* out_attr) { TF_Buffer* attr_buf = TF_NewBuffer(); TF_FunctionGetAttrValueProto(func_, attr_name, attr_buf, s_); ASSERT_TRUE(out_attr->ParseFromArray(attr_buf->data, attr_buf->length)); TF_DeleteBuffer(attr_buf); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 22:08:54 UTC 2023 - 63.6K bytes - Viewed (0) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
} return result; } // Helper method that return list of string for all the StringAttr in the // Attribute identified by 'attr_name'. std::vector<std::string> GetStringsFromDictionaryAttr( const llvm::SmallVector<mlir::DictionaryAttr, 4>& dict_attrs, const StringRef attr_name) { std::vector<std::string> result; for (const auto& arg_attr : dict_attrs) { if (!arg_attr) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
// That will require setting up the tensorflow build with gmock. #define EXPECT_TF_META(attr_name, expected_list_size, expected_type, \ expected_total_size) \ do { \ auto m = TF_OperationGetAttrMetadata(oper, attr_name, s_); \ EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0)