Searched refs:shape_inference_fn (Results 1 – 14 of 14) sorted by relevance
45 : op_def(def), shape_inference_fn(fn), is_function_op(is_function) {} in op_def()48 OpShapeInferenceFn shape_inference_fn; member
67 if (op_reg_data->shape_inference_fn == nullptr) { in InferShapes()73 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in InferShapes()
626 if (op_reg_data_.shape_inference_fn != nullptr) { in SetShapeFn()630 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()
59 *shape_fn_out = op_reg_data.shape_inference_fn; in ExpectSuccess()
1006 ASSERT_NE(op_reg_data->shape_inference_fn, nullptr); in TEST()
29 EXPECT_TRUE(op_reg_data.shape_inference_fn != nullptr) in TEST()
188 op_reg_data->shape_inference_fn = shape_inference_function; in RegisterPerTableLoadOpsForAlgorithmBody()284 op_reg_data->shape_inference_fn = shape_inference_function; in RegisterPerTableRetrieveOpsForAlgorithmBody()
209 CHECK(op_reg_data->shape_inference_fn != nullptr); in TEST()215 Status s = c->Run(op_reg_data->shape_inference_fn); in TEST()
204 ASSERT_TRUE(op_reg_data->shape_inference_fn != nullptr); in TEST()205 TF_ASSERT_OK(c.Run(op_reg_data->shape_inference_fn)); in TEST()
247 if (op_reg_data->shape_inference_fn == nullptr && in AddNode()401 if (op_reg_data->shape_inference_fn == nullptr && in UpdateNode()408 if (!op_reg_data->shape_inference_fn) { in UpdateNode()649 if (op_reg_data->shape_inference_fn) { in RunShapeFn()650 TF_RETURN_IF_ERROR(c->Run(op_reg_data->shape_inference_fn)); in RunShapeFn()
68 if (op_reg_data->shape_inference_fn == nullptr) { in RunCppShapeInferenceImpl()133 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in RunCppShapeInferenceImpl()
86 } else if (!op_reg_data->shape_inference_fn) { in PrintNodeInfo()
1582 if (!c->op_data || c->op_data->shape_inference_fn == nullptr) { in InferShapes()1589 c->inference_context->Run(c->op_data->shape_inference_fn)); in InferShapes()
1883 if (op_reg_data->shape_inference_fn == nullptr) { in CopyShapeInferenceNodeToGraph()