Searched refs:channel_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 103 int channel_dim = input_shape.dims() - 1; in Compile() local
104 int64 channels = input_shape.dim_size(channel_dim); in Compile()
115 /*dimno=*/channel_dim); in Compile()
118 /*dimno=*/channel_dim); in Compile()
121 /*dimno=*/channel_dim); in Compile()
123 channel_shape.set_dim(channel_dim, 1); in Compile()
127 context->SetOutput(0, xla::ConcatInDim(b, hsv, channel_dim)); in Compile()
141 int channel_dim = input_shape.dims() - 1; in Compile() local
142 int64 channels = input_shape.dim_size(channel_dim); in Compile()
152 /*dimno=*/channel_dim); in Compile()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 55 int32 channel_dim = value_tensor.dims() - 1; in GetBiasValueDims() local
56 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim)); in GetBiasValueDims()
57 for (int32 i = 0; i < channel_dim; i++) { in GetBiasValueDims()
112 size_t channel_dim; in Compute() local
114 channel_dim = 1; // NCHW always have channel dim in 1 (with 3, 4, 5 in Compute()
117 channel_dim = input.shape().dims() - 1; // End of code by intel_tf. in Compute()
122 bias.shape().dim_size(0) == input.shape().dim_size(channel_dim), in Compute()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 2395 DimensionHandle channel_dim = in __anon3e672dd83e02() local
2402 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(vec, 0), &channel_dim)); in __anon3e672dd83e02()
2407 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 3, channel_dim, &y)); in __anon3e672dd83e02()
2409 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 1, channel_dim, &y)); in __anon3e672dd83e02()
2412 ShapeHandle vector_shape = c->Vector(channel_dim); in __anon3e672dd83e02()
2462 DimensionHandle channel_dim = (data_format == "NHWC") in __anon3e672dd83f02() local
2466 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(x, 3), &channel_dim)); in __anon3e672dd83f02()
2468 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(x, 1), &channel_dim)); in __anon3e672dd83f02()
2475 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(vec, 0), &channel_dim)); in __anon3e672dd83f02()
2481 c->ReplaceDim(y_backprop, 3, channel_dim, &x_backprop)); in __anon3e672dd83f02()
[all …]
|
| 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()
|
| /external/tensorflow/tensorflow/contrib/image/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()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 804 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormShape() local
810 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(vec, 0), &channel_dim)); in FusedBatchNormShape()
814 TF_RETURN_IF_ERROR(c->ReplaceDim(x, channel_dim_index, channel_dim, &y)); in FusedBatchNormShape()
816 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape()
840 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape() local
842 c->Merge(channel_dim, c->Dim(x, channel_dim_index), &channel_dim)); in FusedBatchNormGradShape()
848 TF_RETURN_IF_ERROR(c->Merge(channel_dim, c->Dim(vec, 0), &channel_dim)); in FusedBatchNormGradShape()
853 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
855 c->set_output(1, c->Vector(channel_dim)); in FusedBatchNormGradShape()
856 c->set_output(2, c->Vector(channel_dim)); in FusedBatchNormGradShape()
[all …]
|