Searched refs:GetTensorInnerFeatureDimIndex (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | tensor_format.h | 211 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex() function
546 dim_sizes[GetTensorInnerFeatureDimIndex(dims, format)] = 4; in ShapeFromFormat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 466 shape_handle, GetTensorInnerFeatureDimIndex(num_dims, tensor_format)); in CheckFormatConstraintsOnShape()
507 dims_actual[GetTensorInnerFeatureDimIndex(num_dims, format)] = in MakeShapeFromFormat()
542 context->Dim(shape, GetTensorInnerFeatureDimIndex(rank, format)), in DimensionsFromShape()
574 out_dims[GetTensorInnerFeatureDimIndex(rank, format)] = *vect_size; in ShapeFromDimensions()
747 GetTensorInnerFeatureDimIndex(rank, data_format))); in Conv2DShapeImpl()
|