HomeSort by relevance Sort by last modified time
    Searched refs:ConvOpAttrs (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_op_helpers.h 38 // ConvOpAttrs contains all of the metadata necessary to specify a TF or XLA
40 struct ConvOpAttrs {
41 // Constructs a ConvOpAttrs, reading most of the attributes from `ctx`.
42 static xla::StatusOr<ConvOpAttrs> Create(int num_spatial_dims, bool depthwise,
59 const ConvOpAttrs& attrs);
62 xla::XlaOp out_backprop, const ConvOpAttrs& attrs);
66 const ConvOpAttrs& attrs);
conv_ops.cc 47 xla::StatusOr<ConvOpAttrs> attrs =
48 ConvOpAttrs::Create(num_spatial_dims, depthwise, ctx);
61 ConvOpAttrs attrs_;
94 xla::StatusOr<ConvOpAttrs> attrs =
95 ConvOpAttrs::Create(num_spatial_dims, depthwise, ctx);
114 ConvOpAttrs attrs_;
152 xla::StatusOr<ConvOpAttrs> attrs =
153 ConvOpAttrs::Create(num_spatial_dims, depthwise, ctx);
172 ConvOpAttrs attrs_;
conv_op_helpers.cc 165 // Performs some basic checks on ConvOpAttrs that are true for all kinds of XLA
167 Status CheckConvAttrs(const ConvOpAttrs& attrs) {
222 xla::StatusOr<ConvOpAttrs> ConvOpAttrs::Create(int num_spatial_dims,
225 ConvOpAttrs attrs;
248 const ConvOpAttrs& attrs) {
325 xla::XlaOp out_backprop, const ConvOpAttrs& attrs) {
397 const ConvOpAttrs& attrs) {

Completed in 228 milliseconds