Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.h273 inline int GetFilterTensorInputChannelsDimIndex(int num_dims, in GetFilterTensorInputChannelsDimIndex() function
572 dim_sizes[GetFilterTensorInputChannelsDimIndex(dims, format)] = I; in ShapeFromFilterTensorFormat()
H A Dtensor_format_test.cc196 EXPECT_EQ(GetFilterTensorInputChannelsDimIndex(num_dims, format), fdm.i()); in RunDimensionIndexesTest()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc1687 int idx = tensorflow::GetFilterTensorInputChannelsDimIndex( in Verify()