Searched refs:SpatialArray (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 86 using SpatialArray = gtl::InlinedVector<int64, 3>;
87 ConvParameters(int64 batch, int64 in_depths, const SpatialArray& in, in ConvParameters()
89 const SpatialArray& filter, const SpatialArray& dilation, in ConvParameters()
90 const SpatialArray& stride, const SpatialArray& padding, in ConvParameters()
159 std::tuple<int64, int64, SpatialArray, TensorFormat, int64, SpatialArray,
160 SpatialArray, SpatialArray, SpatialArray, DataType, int>;
173 static const SpatialArray& CheckSpatialArraySize(const SpatialArray& array) { in CheckSpatialArraySize()
194 SpatialArray in_;
196 SpatialArray filter_;
197 SpatialArray dilation_;
[all …]
|
| D | bias_op.cc | 425 using SpatialArray = gtl::InlinedVector<int64, 4>; typedef in tensorflow::BiasAddParams
426 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
457 using ParamsDataType = std::tuple<SpatialArray, TensorFormat, DataType, int>;
466 SpatialArray in_shape_;
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv_ops_gpu.h | 31 FusedConvParameters(int64 batch, int64 in_depths, const SpatialArray& in, in FusedConvParameters()
33 const SpatialArray& filter, const SpatialArray& dilation, in FusedConvParameters()
34 const SpatialArray& stride, const SpatialArray& padding, in FusedConvParameters()
|