- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TF_GraphGetTensorNumDims (0.21 sec)
-
tensorflow/c/eager/c_api_unified_experimental_graph.cc
} tensorflow::Status Shape( tensorflow::PartialTensorShape* shape) const override { DCHECK(shape != nullptr); TF_Status status; int num_dims = TF_GraphGetTensorNumDims(graph_, output_, &status); DCHECK_GE(num_dims, -1); TF_RETURN_IF_ERROR(StatusFromTF_Status(&status)); if (num_dims == kUnknownRank) { return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
num_dims = TF_GraphGetTensorNumDims(graph, feed_out_0, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); EXPECT_EQ(-1, num_dims); // Set the shape to be unknown, expect no change. TF_GraphSetTensorShape(graph, feed_out_0, /*dims=*/nullptr, -1, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); num_dims = TF_GraphGetTensorNumDims(graph, feed_out_0, s);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} tensorflow::shape_inference::ShapeHandle new_shape = tensorflow::ShapeHandleFromDims(ic, num_dims, dims); status->status = graph->refiner.SetShape(node, output.index, new_shape); } int TF_GraphGetTensorNumDims(TF_Graph* graph, TF_Output output, TF_Status* status) { Node* node = &output.oper->node; mutex_lock l(graph->mu); tensorflow::shape_inference::InferenceContext* ic =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
RELEASE.md
promoted the non-batch version of the ops to handle batches of matrices. * Tracing/timeline support for distributed runtime (no GPU profiler yet). * C API gives access to inferred shapes with `TF_GraphGetTensorNumDims` and `TF_GraphGetTensorShape`. * Shape functions for core ops have moved to C++ via `REGISTER_OP(...).SetShapeFn(...)`. Python shape inference RegisterShape
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0)