saved_model_aot_test.cc
imports tensorflow/core/platform/protobuf.himports tsl/platform/errors.h
SHA-2569f273216f89eac624cea7e74414d21c1f8a21673a5088e2c2483ba0ccf4138f9
MaleculeMd₂(In₃Pa)
Evidence
1/* Copyright 2023 The TensorFlow Authors. All Rights Reserved.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is …
11:42… IND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14================================================================ …
14:73… ======*/
15#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
16
17#include <gtest/gtest.h>
18#include "absl/status/statusor.h"
19#include "absl/strings/string_view.h"
20#include "tensorflow/cc/framework/scope.h"
21#include "tensorflow/cc/ops/function_ops.h"
22#include "tensorflow/cc/ops/math_ops.h"
23#include "tensorflow/compiler/jit/test_util.h"
24#include "tensorflow/compiler/tf2xla/xla_compiler.h"
25#include "tensorflow/core/framework/function.h"
26#include "tensorflow/core/framework/graph_to_functiondef.h"
27#include "tensorflow/core/framework/op.h"
28#include "tensorflow/core/framework/tensor_shape.h"
29#include "tensorflow/core/framework/types.h"
30#include "tensorflow/core/platform/protobuf.h"
31#include "tensorflow/core/tfrt/saved_model/saved_model_aot_compile.h"
32#include "tsl/platform/errors.h"
33#include "tsl/platform/statusor.h"
34
35namespace tensorflow {
⋯5 lines
123:2… &compilation_result_ptr));
124 EXPECT_TRUE(compilation_result.computation != nullptr);
125 EXPECT_TRUE(pjrt_executable != nullptr);
126}
127
128TEST(SavedModelAotTest, AotCompileTfF …