Searched refs:flattened_size (Results 1 – 1 of 1) sorted by relevance
830 int32 flattened_size = 1; in ConvertSoftmaxOperator() local832 flattened_size *= input_shape.dims(i); in ConvertSoftmaxOperator()835 flattened_size, input_shape.dims(input_shape.dimensions_count() - 1)}; in ConvertSoftmaxOperator()871 int32 flattened_size = 1; in ConvertLogSoftmaxOperator() local873 flattened_size *= input_shape.dims(i); in ConvertLogSoftmaxOperator()876 flattened_size, input_shape.dims(input_shape.dimensions_count() - 1)}; in ConvertLogSoftmaxOperator()