Package com.viam.sdk.core.service.mlmodel

Class MLModelRPCClient

java.lang.Object

com.viam.sdk.core.resource.Resource

com.viam.sdk.core.service.Service

com.viam.sdk.core.service.mlmodel.MLModel

com.viam.sdk.core.service.mlmodel.MLModelRPCClient

public class MLModelRPCClient
extends MLModel

Nested Class Summary

Nested classes/interfaces inherited from class com.viam.sdk.core.resource.Resource

Resource.Reconfigurable, Resource.Stoppable

Field Summary

Fields inherited from class com.viam.sdk.core.service.mlmodel.MLModel

SUBTYPE

Constructor Summary

Constructors

Constructor	Description
MLModelRPCClient(String name, Channel chan)

Method Summary

Modifier and Type	Method	Description
Metadata	getMetadata(Optional<com.google.protobuf.Struct> extra)
Map<String,ai.djl.ndarray.NDArray>	infer(Map<String,ai.djl.ndarray.NDArray> inputTensors, ai.djl.ndarray.NDManager manager, Optional<com.google.protobuf.Struct> extra)

Methods inherited from class com.viam.sdk.core.service.mlmodel.MLModel

doCommand, fromRobot, named

Methods inherited from class com.viam.sdk.core.resource.Resource

close, createStatus, getName, getSubtype, named

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

MLModelRPCClient

public MLModelRPCClient(String name,
 Channel chan)

Method Details

infer

public Map<String,ai.djl.ndarray.NDArray> infer(Map<String,ai.djl.ndarray.NDArray> inputTensors,
 ai.djl.ndarray.NDManager manager,
 Optional<com.google.protobuf.Struct> extra)

Specified by:

infer in class MLModel

getMetadata

public Metadata getMetadata(Optional<com.google.protobuf.Struct> extra)

Specified by:

getMetadata in class MLModel