- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for GetNodeAttr (0.3 sec)
-
tensorflow/cc/gradients/nn_grad.cc
TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "dilations", &dilations)); TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "strides", &strides)); TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "explicit_paddings", &explicit_paddings)); TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "data_format", &data_format)); TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "padding", &padding));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 27 23:34:33 UTC 2022 - 24.5K bytes - Viewed (0) -
tensorflow/cc/gradients/image_grad.cc
std::vector<Output>* grad_outputs) { bool align_corners; TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "align_corners", &align_corners)); bool half_pixel_centers; TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "half_pixel_centers", &half_pixel_centers));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 11 00:29:23 UTC 2021 - 5.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
int64_t new_axis_mask; int64_t shrink_axis_mask; TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "begin_mask", &begin_mask)); TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "end_mask", &end_mask)); TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "ellipsis_mask", &ellipsis_mask)); TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "new_axis_mask", &new_axis_mask)); TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
int tpu_core; TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "tpu_core", &tpu_core)); EXPECT_EQ(tpu_core, 1); TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "tpu_core", &tpu_core)); EXPECT_EQ(tpu_core, 0); // Check XlaHostCompute nodes' "shapes" attr. "0" should not have shapes, and // "1" should have shapes. std::vector<TensorShapeProto> shapes; TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "shapes", &shapes));
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/extract_outside_compilation_pass.cc
recv_at_host_dtypes->resize(arg_nodes.size(), DT_INVALID); for (auto* n : arg_nodes) { int index; TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); DataType dtype; TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "T", &dtype)); (*recv_at_host_dtypes)[index] = dtype; } for (int i = 0, end = recv_at_host_dtypes->size(); i < end; i++) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
auto node_name_index = g->BuildNodeNameIndex(); for (auto n : placeholder_nodes) { string node_name; int node_src_output; TF_RETURN_IF_ERROR(GetNodeAttr( n->attrs(), kOutsideCompilationOriginalNodeAttrName, &node_name)); TF_RETURN_IF_ERROR(GetNodeAttr( n->attrs(), kOutsideCompilationSrcOutputAttrName, &node_src_output)); auto iter = node_name_index.find(node_name);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/cc/gradients/functional_grad.cc
const std::vector<Output>& grad_inputs, 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;
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/build_xla_ops_pass.cc
} Status GetXlaClusterInfo(Node* n, XlaClusterInfo* result) { int num_constant_inputs, num_resource_inputs; TF_RETURN_IF_ERROR( GetNodeAttr(n->attrs(), kXlaNumConstantArgsAttr, &num_constant_inputs)); TF_RETURN_IF_ERROR( GetNodeAttr(n->attrs(), kXlaNumResourceArgsAttr, &num_resource_inputs)); if (num_constant_inputs < 0 || num_resource_inputs < 0 ||
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/jit/shape_inference.cc
// And Args must have `index` attribute. TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); } else if (n->type_string() == "Placeholder") { // Use custom attribute (prefixed with `_`) `_index` for placeholders as // they come from user specifications. if (const auto s = GetNodeAttr(n->attrs(), "_index", &index); !s.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
return false; } return guaranteed_constant; } // Finds the `index` of an _Arg or _Retval node. Status GetIndexAttr(const Node& n, int num_args, int* index) { TF_RETURN_IF_ERROR(GetNodeAttr(n.attrs(), "index", index)); if (*index < 0 || *index >= num_args) { return errors::InvalidArgument("Invalid ", n.type_string(), " number ", *index); } return absl::OkStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0)