- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 40 for set_b (0.16 sec)
-
tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc
{"Tperm", DT_INT32}, }}}, {{"c", "t0:y:0"}}); AttrValue true_attribute; true_attribute.set_b(true); (*called_func.mutable_attr())[kXlaMustCompileAttr] = true_attribute; *library.add_function() = called_func; TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator_test.cc
std::unique_ptr<FunctionLibraryDefinition> lib_def_; std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; std::unique_ptr<OpKernel> kernel_; }; AttrValue BoolAttr(bool b) { AttrValue v; v.set_b(b); return v; } TEST_F(XlaKernelCreatorTest, OneFloatOneResourceArgument) { FunctionDef fdef = XTimesY(); (*fdef.mutable_attr())["_XlaMustCompile"] = BoolAttr(true); Init({fdef});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/jit/node_matchers.cc
return props; } std::pair<string, AttrValue> impl::AttrLiteralHelper( const std::pair<string, bool>& bool_attr) { AttrValue attr_value; attr_value.set_b(bool_attr.second); return {bool_attr.first, attr_value}; } std::pair<string, AttrValue> impl::AttrLiteralHelper( const std::pair<string, absl::Span<const int>>& int_list_attr) { AttrValue attr_value;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 03 16:15:20 UTC 2022 - 16.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
Status ConvertAttribute(const mlir::BoolAttr& attr, AttrValue* value) { value->set_b(attr.getValue()); return absl::OkStatus(); } Status ConvertAttribute(const mlir::IntegerAttr& attr, AttrValue* value) { value->set_i(attr.getInt()); return absl::OkStatus(); } Status ConvertAttribute(const mlir::FloatAttr& attr, AttrValue* value) { value->set_f(attr.getValueAsDouble()); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util_test.cc
GraphDefBuilder b(GraphDefBuilder::kFailImmediately); Scope root = Scope::NewRootScope().ExitOnError(); FunctionDefLibrary library; AttrValue true_attribute; true_attribute.set_b(true); FunctionDef identity_func = FunctionDefHelper::Create( "IdentityFunc", /*in_def=*/{"x:float"}, /*out_def=*/{"res:float"}, /*attr_def=*/{},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 22.3K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); EXPECT_EQ(1, val); TF_DeleteStatus(status); return static_cast<void*>(s); }; AttrValue v; v.set_b(true); CreateAndCallKernelWithAttr(my_create_func, "TestKernelAttrBool", v); } TEST_F(TestKernelAttr, BoolList) { auto my_create_func = [](TF_OpKernelConstruction* ctx) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
internal/logger/target/testlogger/testlogger.go
// Call the returned function to disable logging. func (t *testLogger) SetLogTB(tb testing.TB) func() { return t.setTB(tb, logMessage) } // SetErrorTB will set the logger to output to tb.Error. // Call the returned function to disable logging. func (t *testLogger) SetErrorTB(tb testing.TB) func() { return t.setTB(tb, errorMessage) } // SetFatalTB will set the logger to output to tb.Panic.
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Fri May 24 23:05:23 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc
(*node_def.mutable_attr())["shared_name"].set_s(node_def.name()); } else { // Use the concat of function name and node name for such ops in a // function as the shared_name. "@" is used as the separator because // it is not allowed in the function name or the node name. (*node_def.mutable_attr())["shared_name"].set_s( absl::StrCat(node_def.name(), "@", func_name));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 14:33:47 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/tensor_array_side_effect_analysis.h
// Return if the function contains only non-side-effecting ops or TensorArray // ops. bool HasAtMostTensorArrayEffect(mlir::func::FuncOp func_op) const { return set_.count(func_op) > 0; } private: llvm::DenseSet<mlir::func::FuncOp> set_; }; } // namespace tfrt_compiler } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/tensor_array_side_effect_analysis.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 01:49:07 UTC 2022 - 1.8K bytes - Viewed (0)