Searched refs:inference_context (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | test_util.cc | 110 std::unique_ptr<InferenceContext> inference_context; in Invoke() local
112 runtime_options, &objects, command_queue.get(), &inference_context)); in Invoke()
113 RETURN_IF_ERROR(inference_context->Reset()); in Invoke()
116 RETURN_IF_ERROR(inference_context->Execute()); in Invoke()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 634 std::unique_ptr<InferenceContext> inference_context; member
649 auto* ic = inference_context.get(); in StringifyShapeHandle()
668 auto* ic = inference_context.get(); in DebugString()
743 return it->second.inference_context.get(); in GetContext()
773 auto* ic = ctx->inference_context.get(); in UpdateFunction()
869 auto* ic = ctx->inference_context.get(); in UpdateFunction()
970 InferenceContext* ic = ctx->inference_context.get(); in UpdateNode()
986 InferenceContext* src_ic = src_ctx->inference_context.get(); in UpdateNode()
1339 node_ctx.inference_context.reset(new InferenceContext( in AddNode()
1343 const Status s = node_ctx.inference_context->construction_status(); in AddNode()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | BUILD | 35 "//tensorflow/lite/delegates/gpu/cl:inference_context",
51 "//tensorflow/lite/delegates/gpu/cl:inference_context",
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 545 shape_inference::InferenceContext* const inference_context, in InferenceShape() argument
555 dims_handle.push_back(inference_context->MakeDim(dims)); in InferenceShape()
557 *shape_handle = inference_context->MakeShape(dims_handle); in InferenceShape()
561 *shape_handle = inference_context->UnknownShape(); in InferenceShape()
567 dims_handle[i] = inference_context->MakeDim(inference_shape_.dim_size(i)); in InferenceShape()
569 *shape_handle = inference_context->MakeShape(dims_handle); in InferenceShape()
575 shape_inference::InferenceContext* const inference_context, in SetInferenceShape() argument
577 auto num_dims = inference_context->Rank(shape_handle); in SetInferenceShape()
586 dims[i] = inference_context->Value(inference_context->Dim(shape_handle, i)); in SetInferenceShape()
|
| D | tensor_handle.h | 233 shape_inference::InferenceContext* const inference_context,
236 shape_inference::InferenceContext* const inference_context,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | BUILD | 143 name = "inference_context",
144 srcs = ["inference_context.cc"],
145 hdrs = ["inference_context.h"],
340 "//tensorflow/lite/delegates/gpu/metal:inference_context",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | api.h | 57 std::unique_ptr<InferenceContext>* inference_context) const = 0;
|
| D | api.cc | 248 std::unique_ptr<InferenceContext>* inference_context) const final { in NewRun()
271 *inference_context = absl::make_unique<InferenceContextWithBatchImpl>( in NewRun()
274 *inference_context = in NewRun()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | BUILD | 32 ":inference_context",
379 name = "inference_context",
381 "inference_context.cc",
385 "inference_context.h",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | BUILD | 102 "//tensorflow/lite/delegates/gpu/metal:inference_context",
|
| D | metal_delegate.mm | 45 #include "tensorflow/lite/delegates/gpu/metal/inference_context.h"
|