| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| image_ops.cc | 91 int channel_dim = input_shape.dims() - 1; variable
92 int64 channels = input_shape.dim_size(channel_dim);
103 /*dimno=*/channel_dim);
106 /*dimno=*/channel_dim);
109 /*dimno=*/channel_dim);
111 channel_shape.set_dim(channel_dim, 1);
115 context->SetOutput(0, b->ConcatInDim(hsv, channel_dim));
129 int channel_dim = input_shape.dims() - 1; variable
130 int64 channels = input_shape.dim_size(channel_dim);
140 /*dimno=*/channel_dim);
169 int channel_dim = input_shape.dims() - 1; variable
213 const int channel_dim = input_shape.dims() - 1; variable
263 const int channel_dim = input_shape.dims() - 1; variable
[all...] |
| /external/tensorflow/tensorflow/python/keras/_impl/keras/applications/ |
| nasnet.py | 212 channel_dim = 1 if K.image_data_format() == 'channels_first' else -1
235 axis=channel_dim, momentum=0.9997, epsilon=1e-3, name='stem_bn1')(
482 channel_dim = 1 if K.image_data_format() == 'channels_first' else -1
496 axis=channel_dim,
511 axis=channel_dim,
533 channel_dim = 1 if K.image_data_format() == 'channels_first' else -1
579 p = concatenate([p1, p2], axis=channel_dim)
581 axis=channel_dim,
587 elif p_shape[channel_dim] != filters:
599 axis=channel_dim,
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| bias_op.cc | 52 int32 channel_dim = value_tensor.dims() - 1; local
53 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim));
54 for (int32 i = 0; i < channel_dim; i++) {
58 int32 channel_dim = value_tensor.dims() - 3; local
61 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim));
64 for (int32 i = 0; i < channel_dim; i++) {
109 size_t channel_dim; variable
114 channel_dim = 1;
116 channel_dim = input.shape().dims() - 1; // End of code by intel_tf.
121 bias.shape().dim_size(0) == input.shape().dim_size(channel_dim),
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 705 DimensionHandle channel_dim = local
712 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(vec, 0), &channel_dim));
717 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 3, channel_dim, &y));
719 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 1, channel_dim, &y));
722 ShapeHandle vector_shape = c->Vector(channel_dim);
740 DimensionHandle channel_dim = local
743 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(x, 3), &channel_dim));
745 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(x, 1), &channel_dim))
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| nn_ops.cc | [all...] |
| image_ops.cc | 28 // Sets output[0] to shape [batch_dim,height,width,channel_dim], where
31 int size_input_idx, DimensionHandle channel_dim) {
58 c->set_output(0, c->MakeShape({batch_dim, height, width, channel_dim}));
|
