- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 185 for nodedef (0.12 sec)
-
tensorflow/c/kernels/summary_op_test.cc
} void TestScalarSummaryOp(Tensor* tags, Tensor* values, string expected_output, error::Code expected_code) { // Initialize node used to fetch OpKernel Status status; NodeDef def; def.set_op("ScalarSummary"); def.set_device(DEVICE_CPU); AttrValue valuesTypeAttr; SetAttrValue(values->dtype(), &valuesTypeAttr); (*def.mutable_attr())["T"] = valuesTypeAttr;
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/jit/clone_constants_for_better_clustering.cc
} while (name_set.contains(candidate)); return candidate; } absl::StatusOr<Node*> CloneConstantsForBetterClusteringPassImpl::CloneNode( const absl::flat_hash_set<string>& name_set, Node* n) { NodeDef new_in_def = n->def(); new_in_def.clear_input(); new_in_def.set_name(GenerateUniqueName(name_set, new_in_def.name())); TF_ASSIGN_OR_RETURN(Node * new_in, graph_->AddNode(new_in_def));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.cc
std::find(node.output_types().begin(), node.output_types().end(), DT_RESOURCE) != node.output_types().end(); } void RemoveFromXlaCluster(NodeDef* node_def) { node_def->mutable_attr()->erase(kXlaClusterAttr); } void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); } namespace { typedef xla_config_registry::XlaGlobalJitLevel XlaGlobalJitLevel;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op_test.cc
.Input(inputs) .Attr("ids", ids) .Attr("calibration_methods", calibration_methods) .Finalize(node_def())); ASSERT_THAT(InitOp(), StatusIs(tsl::error::INVALID_ARGUMENT, HasSubstr("NodeDef missing attr 'output_file_path'"))); } TEST_F(CalibrationStatisticsSaverTest, WrongNumInputs) { std::vector<std::string> ids{"1"};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 01:31:23 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
const FunctionLibraryDefinition& flib_def) { for (const std::string& func_name : flib_def.ListFunctionNames()) { const FunctionDef* func_def = flib_def.Find(func_name); for (const NodeDef& node_def : func_def->node_def()) { if (node_def.op() == kPartitionedCall) return true; } } return false; } } // namespace bool IsSupportedByNonReplicatedBridge(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model_test.cc
// In both cases, none should be variables. size_t expected_nodes = use_resource ? (use_identity ? 5 : 4) : 3; EXPECT_EQ(frozen_graph_def.node_size(), expected_nodes); for (const NodeDef& node : frozen_graph_def.node()) { EXPECT_NE(node.op(), "Variable") << node.name(); EXPECT_NE(node.op(), "VariableV2") << node.name(); EXPECT_NE(node.op(), "VarHandleOp") << node.name();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 13:30:31 UTC 2022 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/auto_clustering_test_helper.cc
return result; } Status AssertGraphDefIsUnclustered(const GraphDef& graphdef) { const char* kXlaClusterAttr = "_XlaCluster"; const char* kXlaAlreadyClusteredAttr = "_XlaAlreadyClustered"; for (const NodeDef& node : graphdef.node()) { if (node.attr().count(kXlaClusterAttr) || node.attr().count(kXlaAlreadyClusteredAttr)) { return errors::InvalidArgument(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 12:11:46 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/jit/partially_decluster_pass_test.cc
TF_ASSERT_OK(DeviceFactory::AddDevices( SessionOptions(), "/job:localhost/replica:0/task:0", &devices)); // Scope::ToGraph loses the assigned device name since it goes through // GraphDef/NodeDef which does not have a field for the assigned device name. Node* n = FindNodeByName(*graph, "shape"); ASSERT_NE(n, nullptr); n->set_assigned_device_name( "/job:localhost/replica:0/task:0/device:XLA_GPU:0");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 23K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
} Status MakeXlaCompiledKernel(Graph* graph, const string& callee_name, const string& node_name, int num_constant_args, int num_resource_args, Node** result) { NodeDef call_node; call_node.set_name(node_name); call_node.set_op(callee_name); AddNodeAttr(kXlaCompiledKernelAttr, true, &call_node); AddNodeAttr(kXlaNumConstantArgsAttr, num_constant_args, &call_node);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
std::initializer_list<int64_t> dims = {2, 3, 4, 5}; Tensor tensor(data_type, TensorShape(dims)); for (int i = 0; i < 2 * 3 * 4 * 5; ++i) { tensor.flat<int32>()(i) = i; } NodeDef node; auto builder = NodeDefBuilder("some_node", "Const") .Attr("dtype", data_type) .Attr("value", tensor); // Create a bad output shape attr. AttrValue shape_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K bytes - Viewed (0)