Searched refs:shape_inference_fn (Results 1 – 18 of 18) sorted by relevance
37 if (op_reg_data->shape_inference_fn == nullptr) return Status::OK(); in RunShapeInference()48 TF_RETURN_IF_ERROR(ic.Run(op_reg_data->shape_inference_fn)); in RunShapeInference()
45 : op_def(def), shape_inference_fn(fn), is_function_op(is_function) {} in op_def()48 OpShapeInferenceFn shape_inference_fn; member
66 if (op_reg_data->shape_inference_fn == nullptr) { in InferShapes()72 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in InferShapes()
629 if (op_reg_data_.shape_inference_fn != nullptr) { in SetShapeFn()633 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()
59 *shape_fn_out = op_reg_data.shape_inference_fn; in ExpectSuccess()
1007 ASSERT_NE(op_reg_data->shape_inference_fn, nullptr); in TEST()
117 TF_CHECK_OK(reg->shape_inference_fn(&c)); in TEST()136 TF_CHECK_OK(reg->shape_inference_fn(&c)); in TEST()154 TF_CHECK_OK(reg->shape_inference_fn(&c)); in TEST()
29 EXPECT_TRUE(op_reg_data.shape_inference_fn != nullptr) in TEST()
41 TF_ASSERT_OK(c.Run(op_reg_data->shape_inference_fn)); in TEST()
230 CHECK(op_reg_data->shape_inference_fn != nullptr); in TEST()236 Status s = c->Run(op_reg_data->shape_inference_fn); in TEST()
237 ASSERT_TRUE(op_reg_data->shape_inference_fn != nullptr); in TEST()238 TF_ASSERT_OK(c.Run(op_reg_data->shape_inference_fn)); in TEST()
238 if (op_reg_data->shape_inference_fn == nullptr && in AddNodeInternal()379 if (op_reg_data->shape_inference_fn == nullptr && in UpdateNode()386 if (!op_reg_data->shape_inference_fn) { in UpdateNode()692 if (op_reg_data->shape_inference_fn) { in RunShapeFn()693 TF_RETURN_IF_ERROR(c->Run(op_reg_data->shape_inference_fn)); in RunShapeFn()
201 if (!op_reg_data->shape_inference_fn) { in InferReturnTypeComponentsForTFOp()261 auto status = c.Run(op_reg_data->shape_inference_fn); in InferReturnTypeComponentsForTFOp()296 auto status = c.Run(op_reg_data->shape_inference_fn); in InferReturnTypeComponentsForTFOp()
86 } else if (!op_reg_data->shape_inference_fn) { in PrintNodeInfo()
556 if (node_data->op_reg_data()->shape_inference_fn == nullptr) { in ValidateOutputTensorShapeConsistency()561 status = c.Run(node_data->op_reg_data()->shape_inference_fn); in ValidateOutputTensorShapeConsistency()
674 if (op_reg_data->shape_inference_fn == nullptr) { in TFE_InferShapes()681 status->status = c.Run(op_reg_data->shape_inference_fn); in TFE_InferShapes()
1882 if (!c->op_data || c->op_data->shape_inference_fn == nullptr || in InferShapes()1883 !c->inference_context->Run(c->op_data->shape_inference_fn).ok()) { in InferShapes()
1077 if (op_reg_data->shape_inference_fn == nullptr) { in InferOutputType()1086 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in InferOutputType()