Home
last modified time | relevance | path

Searched refs:shape_inference_fn (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc37 if (op_reg_data->shape_inference_fn == nullptr) return OkStatus(); in RunShapeInference()
48 TF_RETURN_IF_ERROR(ic.Run(op_reg_data->shape_inference_fn)); in RunShapeInference()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc117 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()
/external/tensorflow/tensorflow/core/framework/
Dop_def_builder.h69 : op_def(def), shape_inference_fn(fn), is_function_op(is_function) {} in op_def()
72 OpShapeInferenceFn shape_inference_fn; member
Dshape_inference_testutil.cc66 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()
Dop_def_builder.cc653 if (op_reg_data_.shape_inference_fn != nullptr) { in SetShapeFn()
657 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()
Dop_def_builder_test.cc59 *shape_fn_out = op_reg_data.shape_inference_fn; in ExpectSuccess()
Dfunction_test.cc1074 ASSERT_NE(op_reg_data->shape_inference_fn, nullptr); in TEST()
/external/tensorflow/tensorflow/core/ops/
Dshape_function_test.cc29 EXPECT_TRUE(op_reg_data.shape_inference_fn != nullptr) in TEST()
Dfunctional_ops_test.cc41 TF_ASSERT_OK(c.Run(op_reg_data->shape_inference_fn)); in TEST()
Dmath_ops_test.cc230 CHECK(op_reg_data->shape_inference_fn != nullptr); in TEST()
236 Status s = c->Run(op_reg_data->shape_inference_fn); in TEST()
Darray_ops_test.cc237 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()
/external/tensorflow/tensorflow/core/ir/utils/
Dshape_inference_utils.cc216 if (!op_reg_data->shape_inference_fn) { in InferReturnTypeComponentsForTFOp()
281 auto status = c.Run(op_reg_data->shape_inference_fn); in InferReturnTypeComponentsForTFOp()
348 auto status = c.Run(op_reg_data->shape_inference_fn); in InferReturnTypeComponentsForTFOp()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc254 if (op_reg_data->shape_inference_fn == nullptr && in AddNodeInternal()
395 if (op_reg_data->shape_inference_fn == nullptr && in UpdateNode()
402 if (!op_reg_data->shape_inference_fn) { in UpdateNode()
713 if (op_reg_data->shape_inference_fn) { in RunShapeFn()
714 TF_RETURN_IF_ERROR(c->Run(op_reg_data->shape_inference_fn)); in RunShapeFn()
/external/tensorflow/tensorflow/python/grappler/
Dmodel_analyzer.cc86 } else if (!op_reg_data->shape_inference_fn) { in PrintNodeInfo()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc657 if (node_data->op_reg_data()->shape_inference_fn == nullptr) { in ValidateOutputTensorShapeConsistency()
662 status = c.Run(node_data->op_reg_data()->shape_inference_fn); in ValidateOutputTensorShapeConsistency()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc681 if (op_reg_data->shape_inference_fn == nullptr) { in TFE_InferShapes()
688 status->status = c.Run(op_reg_data->shape_inference_fn); in TFE_InferShapes()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlight_outside_compilation.cc145 TF_RETURN_IF_ERROR(data->shape_inference_fn(&ic)); in CompileToCustomCallCallingTfKernel()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1921 if (!c->op_data || c->op_data->shape_inference_fn == nullptr || in InferShapes()
1922 !c->inference_context->Run(c->op_data->shape_inference_fn).ok()) { in InferShapes()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1224 if (op_reg_data->shape_inference_fn == nullptr) { in InferOutputType()
1233 TF_RETURN_IF_ERROR(c.Run(op_reg_data->shape_inference_fn)); in InferOutputType()