Home
last modified time | relevance | path

Searched defs:inference_context (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Dtest_util.cc111 std::unique_ptr<InferenceContext> inference_context; in Invoke() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc734 shape_inference::InferenceContext* inference_context = in PrepareGraphForMlir() local
821 shape_inference::InferenceContext* inference_context = in PrepareGraphForMlir() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dtensor_handle.cc545 shape_inference::InferenceContext* const inference_context, in InferenceShape()
575 shape_inference::InferenceContext* const inference_context, in SetInferenceShape()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc663 std::unique_ptr<InferenceContext> inference_context; member