• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1// This is used for convolution logging. Also see
2// tensorflow/core/protobuf/autotuing.h
3syntax = "proto3";
4
5package xla.gpu;
6
7import "tensorflow/compiler/xla/service/hlo.proto";
8import "tensorflow/compiler/xla/xla_data.proto";
9import "tensorflow/core/protobuf/autotuning.proto";
10
11message ConvInstructionLog {
12  xla.HloInstructionProto instruction = 1;
13  repeated xla.ShapeProto operand_shapes = 2;
14  uint64 result_address = 3;
15  repeated uint64 operand_addresses = 4;
16}
17
18message DenylistedAlgorithm {
19  int64 id = 1;
20  bool tensor_ops = 2;
21}
22
23message AlgorithmDenylistEntry {
24  string hlo = 1;
25  tensorflow.ComputeCapability cc = 2;
26  tensorflow.CudnnVersion cudnn_version = 3;
27  string blas_version = 5;
28  repeated DenylistedAlgorithm algos = 4;
29}
30
31message AlgorithmDenylist {
32  repeated AlgorithmDenylistEntry entries = 1;
33}
34