Searched defs:feature_dim (Results 1 – 6 of 6) sorted by relevance
53 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local98 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local
174 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local270 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local330 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
55 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile() local
128 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local