- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for mutable_attr (0.47 sec)
-
tensorflow/compiler/jit/xla_kernel_creator_test.cc
TEST_F(XlaKernelCreatorTest, OneFloatOneResourceArgument) { FunctionDef fdef = XTimesY(); (*fdef.mutable_attr())["_XlaMustCompile"] = BoolAttr(true); Init({fdef}); XlaKernelCreator xla_kernel_creator; auto callsite = ToNodeProperties(R"pb( name: 'XTimesY' op: 'XTimesY' input: 'a' input: 'b' )pb"); (*(callsite->node_def.mutable_attr()))["_XlaMustCompile"] = BoolAttr(true);
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/c/kernels/bitcast_op_test.cc
def.set_device(DEVICE_CPU); AttrValue typeAttr; SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr); (*def.mutable_attr())["T"] = typeAttr; (*def.mutable_attr())["type"] = outTypeAttr; def.add_input( strings::StrCat("input1: ", DataTypeString(input_tensor->dtype()))); std::unique_ptr<OpKernel> kernel =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K 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(
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/cc/gradients/functional_grad.cc
std::vector<Output>* grad_outputs) { NameAttrList f; TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "f", &f)); for (const auto& attr : op.node()->attrs()) { (*f.mutable_attr())[attr.first] = attr.second; } std::vector<Output> func_inputs; std::vector<DataType> input_dtypes; const int num_inputs = op.num_inputs(); func_inputs.reserve(num_inputs + grad_inputs.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 20:09:06 UTC 2021 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc
}}}, {{"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)); FunctionLibraryDefinition flib_def(OpRegistry::Global(), 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/tests/device_compiler_test_helper.cc
attrs) { NodeDef node; node.set_name(std::string(name)); node.set_op(std::string(op)); for (const auto& input : inputs) node.add_input(input); for (const auto& attr : attrs) node.mutable_attr()->insert({attr.first, attr.second.proto}); return node; } } // namespace GraphDef DeviceCompilerSerializeTest::GetTestGraph( const PartialTensorShape& input_shape) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compilation_cluster_signature_test.cc
} void BM_BuildSignature(::testing::benchmark::State& state) { const int n_args = state.range(0); NameAttrList fn; fn.set_name("afunction"); for (int i = 0; i < n_args; i++) { (*fn.mutable_attr())[absl::StrCat("T", i)].set_type(DT_FLOAT); } std::vector<XlaCompiler::Argument> args(n_args); for (int i = 0; i < n_args; i++) { args[i].kind = (((i % 3) == 0) ? XlaCompiler::Argument::kConstant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/c/kernels/summary_op_test.cc
Status status; NodeDef def; def.set_op("ScalarSummary"); def.set_device(DEVICE_CPU); AttrValue valuesTypeAttr; SetAttrValue(values->dtype(), &valuesTypeAttr); (*def.mutable_attr())["T"] = valuesTypeAttr; def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype()))); def.add_input(strings::StrCat("input2: ", DataTypeString(values->dtype()))); std::unique_ptr<OpKernel> kernel =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc
LOG_FIRST_N(INFO, 1) << "Run Node Expansion Passes"; // Get the FunctionDef and insert that into the context const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); auto& ctx = orig_op->EagerContext(); Fprint128 cache_key = orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName()); // Include soft placement policy in cache key since the placement strategy // can change and thus affect which kernel is picked.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 3.8K bytes - Viewed (0)