- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for kXlaInferredShapesAttrName (0.22 sec)
-
tensorflow/compiler/jit/encapsulate_util_test.cc
auto node_index = g.BuildNodeNameIndex(); Node *add_node = node_index["add"]; std::vector<PartialTensorShape> output_shapes; TF_CHECK_OK(GetNodeAttr(add_node->attrs(), kXlaInferredShapesAttrName, &output_shapes)); EXPECT_EQ(output_shapes.size(), 1); TensorShapeProto shape_proto; output_shapes[0].AsProto(&shape_proto); EXPECT_EQ(shape_proto.dim_size(), 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 05 22:06:22 UTC 2019 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
auto node_name_image = g->BuildNodeNameIndex(); Node *const0_node = node_name_image["const0"]; EXPECT_NE(const0_node, nullptr); PartialTensorShape shape({2}); const0_node->AddAttr(kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); std::vector<OutputTensor> arg_source_tensors;
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/encapsulate_util.h
// Attribute marking output tensor shapes inferred by XLA. Attribute value is // a list of PartialTensorShape objects. extern const char kXlaInferredShapesAttrName[]; // Infers output shapes for all nodes in graph `g`. The output shapes will be // stored in node attribute `kXlaInferredShapesAttrName`. // // We have to perform shape inference before encapsulation because after
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
control_input); } g->AddControlEdge(iter->second, n); } } } return absl::OkStatus(); } } // namespace const char kXlaInferredShapesAttrName[] = "_xla_inferred_shapes"; const char kXlaConnectedToXlaComputationAttrName[] = "_xla_connected_to_xla_computation"; const char kXlaConnectedFromXlaComputationAttrName[] =
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/compiler/jit/extract_outside_compilation_pass.cc
const Edge* e; if (!send_from_host_node->input_edge(i, &e).ok()) { return std::nullopt; } std::vector<PartialTensorShape> shapes; if (!GetNodeAttr(e->src()->attrs(), kXlaInferredShapesAttrName, &shapes) .ok()) { return std::nullopt; } const PartialTensorShape shape = shapes[e->src_output()]; if (!shape.IsFullyDefined()) { return std::nullopt; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)