- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for shape_inference_func (0.21 sec)
-
tensorflow/c/ops.cc
} void TF_OpDefinitionBuilderSetShapeInferenceFunction( TF_OpDefinitionBuilder* builder, void (*shape_inference_func)(TF_ShapeInferenceContext* ctx, TF_Status* status)) { auto* cc_builder = reinterpret_cast<OpDefBuilder*>(builder); cc_builder->SetShapeFn( [shape_inference_func](InferenceContext* ctx) -> tensorflow::Status { TF_Status* c_status = TF_NewStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0) -
tensorflow/c/ops.h
// Sets the shape inference function for the op. TF_CAPI_EXPORT extern void TF_OpDefinitionBuilderSetShapeInferenceFunction( TF_OpDefinitionBuilder* builder, void (*shape_inference_func)(TF_ShapeInferenceContext* ctx, TF_Status* status)); //---------------------------------------------------- // Functions for TF_ShapeInferenceContext. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
std::vector<shape_inference::ShapeHandle> input_shapes; TF_CHECK_OK(c.input("input", &input_shapes)); ASSERT_EQ("[3,4]", c.DebugString(input_shapes[0])); TF_CHECK_OK(reg->shape_inference_fn(&c)); ASSERT_EQ("[3,4,8]", c.DebugString(c.output(0))); } TEST(BitcastOpTest, TestShapeInference_SmallerShape) { const OpRegistrationData* reg;
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/c/c_api_experimental.cc
if (!status->status.ok()) return; if (op_reg_data->shape_inference_fn == nullptr) { status->status = InvalidArgument("No shape inference function exists for op '", node_def.op(), "', did you forget to define it?"); return; } status->status = c.Run(op_reg_data->shape_inference_fn); if (!status->status.ok()) return; // Set output_shapes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
if (!op_reg_data) { DVLOG(1) << "Skipping inference for unregistered op " << node.type_string(); return default_type(); } if (op_reg_data->shape_inference_fn == nullptr) { DVLOG(1) << "Skipping inference for op without shape function " << node.type_string(); return default_type(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)