aws-lambda-docker-serverless-inference/tensorflow-inference-docker-lambda/template.yaml at main · aws-samples/aws-lambda-docker-serverless-inference · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
AWSTemplateFormatVersion: '2010-09-09'
Transform: AWS::Serverless-2016-10-31
Description: >
  tensorflow-inference-docker-lambda

  SAM Template for tensorflow-inference-docker-lambda

Resources:
  TensorFlowInferenceFunction:
    Type: AWS::Serverless::Function # More info about Function Resource: https://github.com/awslabs/serverless-application-model/blob/master/versions/2016-10-31.md#awsserverlessfunction
    Properties:
      PackageType: Image
      MemorySize: 1536
      Timeout: 120
    Metadata:
      DockerTag: python3.7-v1
      DockerContext: .
      Dockerfile: Dockerfile

Outputs:
  TensorFlowInferenceFunction:
    Description: "TensorFlowInference Lambda Function ARN"
    Value: !GetAtt TensorFlowInferenceFunction.Arn
  TensorFlowInferenceFunctionIamRole:
    Description: "Implicit IAM Role created for TensorFlowInference function"
    Value: !GetAtt TensorFlowInferenceFunction.Arn