Searched refs:NCHW (Results 1 – 25 of 41) sorted by relevance
12
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BiasAddGrad.pbtxt | 21 Alternatively, the format could be "NCHW", the data storage order of:
30 For NHWC data format, the feature dimension is the last. For NCHW data format,
|
| D | api_def_FusedBatchNorm.pbtxt | 84 The data format for x and y. Either "NHWC" (default) or "NCHW".
96 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
|
| D | api_def_FusedBatchNormGrad.pbtxt | 87 Either "NHWC" (default) or "NCHW".
99 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
|
| D | api_def_FusedBatchNormV2.pbtxt | 90 The data format for x and y. Either "NHWC" (default) or "NCHW".
102 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 93 Either "NHWC" (default) or "NCHW".
105 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
|
| D | api_def_MaxPool.pbtxt | 40 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_MaxPoolV2.pbtxt | 40 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_BiasAdd.pbtxt | 27 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_AvgPool.pbtxt | 39 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_MaxPoolGradGrad.pbtxt | 52 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_MaxPoolGradGradV2.pbtxt | 52 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_MaxPoolGradV2.pbtxt | 52 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_AvgPoolGrad.pbtxt | 47 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_MaxPoolGrad.pbtxt | 53 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_DepthwiseConv2dNativeBackpropInput.pbtxt | 55 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt | 55 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_DepthwiseConv2dNative.pbtxt | 22 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_Conv2DBackpropFilter.pbtxt | 61 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_Conv2DBackpropInput.pbtxt | 60 Alternatively, the format could be "NCHW", the data storage order of:
|
| D | api_def_SpaceToDepth.pbtxt | 26 "NCHW": `[ batch, channels, height, width ]`
|
| D | api_def_DepthToSpace.pbtxt | 29 "NCHW": `[ batch, channels, height, width ]`
|
| D | api_def_Conv2D.pbtxt | 53 Alternatively, the format could be "NCHW", the data storage order of:
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| D | fused_conv2d_bias_activation_op.cc | 109 "NHWC", "NCHW":
136 "NCHW": `float [ batch, channels, height, width ]`
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.proto | 26 kBatchDepthYX = 3; // cuDNN's NCHW layout
38 kOutputInputYX = 0; // cuDNN's NCHW layout
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_test.cc | 231 BM_BIAS_ADD_GRAD_ALL(gpu, NCHW, float, DT_FLOAT);
232 BM_BIAS_ADD_GRAD_ALL(gpu, NCHW, half, DT_HALF);
|
12