Searched refs:channel_dim (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 105 int channel_dim = input_shape.dims() - 1; in Compile() local
106 int64 channels = input_shape.dim_size(channel_dim); in Compile()
117 /*dimno=*/channel_dim); in Compile()
120 /*dimno=*/channel_dim); in Compile()
123 /*dimno=*/channel_dim); in Compile()
125 channel_shape.set_dim(channel_dim, 1); in Compile()
129 context->SetOutput(0, xla::ConcatInDim(b, hsv, channel_dim)); in Compile()
143 int channel_dim = input_shape.dims() - 1; in Compile() local
144 int64 channels = input_shape.dim_size(channel_dim); in Compile()
154 /*dimno=*/channel_dim); in Compile()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | nasnet.py | 208 channel_dim = 1 if backend.image_data_format() == 'channels_first' else -1
221 axis=channel_dim, momentum=0.9997, epsilon=1e-3, name='stem_bn1')(
492 channel_dim = 1 if backend.image_data_format() == 'channels_first' else -1
513 axis=channel_dim,
528 axis=channel_dim,
550 channel_dim = 1 if backend.image_data_format() == 'channels_first' else -1
593 p = layers.concatenate([p1, p2], axis=channel_dim)
595 axis=channel_dim,
601 elif p_shape[channel_dim] != filters:
613 axis=channel_dim,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 54 int32 channel_dim = value_tensor.dims() - 1; in GetBiasValueDims() local
55 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim)); in GetBiasValueDims()
56 for (int32 i = 0; i < channel_dim; i++) { in GetBiasValueDims()
111 size_t channel_dim; in Compute() local
113 channel_dim = 1; // NCHW always have channel dim in 1 (with 3, 4, 5 in Compute()
116 channel_dim = input.shape().dims() - 1; // End of code by intel_tf. in Compute()
121 bias.shape().dim_size(0) == input.shape().dim_size(channel_dim), in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1153 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormShape() local
1159 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(vec, 0), &channel_dim)); in FusedBatchNormShape()
1163 TF_RETURN_IF_ERROR(c->ReplaceDim(x, channel_dim_index, channel_dim, &y)); in FusedBatchNormShape()
1165 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape()
1193 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormExShape() local
1196 if (c->ValueKnown(channel_dim) && c->Value(channel_dim) % 4 != 0) { in FusedBatchNormExShape()
1223 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape() local
1225 c->Merge(channel_dim, c->Dim(x, channel_dim_index), &channel_dim)); in FusedBatchNormGradShape()
1231 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(vec, 0), &channel_dim)); in FusedBatchNormGradShape()
1236 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 820 const uint64_t channel_dim = rank - 1; in matchAndRewrite() local
822 rw.window_dimensions().getValue<int64_t>({channel_dim}) != 1 || in matchAndRewrite()
823 window_strides[batch_dim] != 1 || window_strides[channel_dim] != 1 || in matchAndRewrite()
825 padding[2 * channel_dim] != 0 || padding[2 * channel_dim + 1] != 0) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights.cc | 50 int channel_dim; member
446 model.get(), tensor_pair.second.t, tensor_pair.second.channel_dim, in QuantizeWeightsInt8()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 31 int size_input_idx, DimensionHandle channel_dim) { in SetOutputToSizedImage() argument
58 c->set_output(0, c->MakeShape({batch_dim, height, width, channel_dim})); in SetOutputToSizedImage()
|