Home
last modified time | relevance | path

Searched refs:GetFilterTensorOutputChannelsDimIndex (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.h288 inline int GetFilterTensorOutputChannelsDimIndex(int num_dims, in GetFilterTensorOutputChannelsDimIndex() function
561 dim_sizes[GetFilterTensorOutputChannelsDimIndex(dims, format)] = O; in ShapeFromFilterTensorFormat()
H A Dtensor_format_test.cc194 EXPECT_EQ(GetFilterTensorOutputChannelsDimIndex(num_dims, format), fdm.o()); in RunDimensionIndexesTest()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc1766 filter_ty.getDimSize(GetFilterTensorOutputChannelsDimIndex( in inferConvReturnTypeComponents()