- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 25 for mutable_attr (0.16 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/compiler/mlir/tensorflow/translate/export_graphdef.cc
} (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model.cc
NodeDef* const_node) { const_node->set_name(variable_node.name()); const_node->set_op("Const"); (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype"); variable_value.AsProtoTensorContent( (*const_node->mutable_attr())["value"].mutable_tensor()); } // Converts a ReadVariableOp NodeDef to an Identity NodeDef. void ConvertReadVariableOpToIdentity(const NodeDef& node,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; std::vector<string> shape_inference_graphs; bool has_outside_compilation; NameAttrList name_attrs; name_attrs.set_name("cluster"); *name_attrs.mutable_attr() = attrs; TF_CHECK_OK(ExtractOutsideCompilationTest( "_xla", "_oc", "cluster", name_attrs, "cluster_rewritten", "host_graph", host_compute_core, &fld, &shape_inference_graphs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K 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/mlir/tensorflow/translate/export_tf_dialect_op.cc
// executed on TPU or GPU devices, which do not have `Cast` registered as a // runtime OpKernel. void RemoveIdentityCast(NodeDef* node_def) { auto attr = node_def->mutable_attr(); if (node_def->op() == "Cast" && attr->contains("SrcT") && attr->contains("DstT") && attr->at("SrcT").type() == attr->at("DstT").type() && attr->contains("Truncate") && !attr->at("Truncate").b()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0)