Searched refs:result_dim (Results 1 – 2 of 2) sorted by relevance
333 int64 result_dim; member370 for (int64 result_dim = result_shape.size() - 1; result_dim >= 0; in ComputeReshapePassthroughDimPairs() local371 --result_dim) { in ComputeReshapePassthroughDimPairs()379 << "result_dim = " << result_dim in ComputeReshapePassthroughDimPairs()385 result_shape[result_dim] == operand_shape[candidate_operand_dim - 1]) { in ComputeReshapePassthroughDimPairs()386 result.push_back({/*result_dim=*/result_dim, in ComputeReshapePassthroughDimPairs()389 result_subarray_size *= result_shape[result_dim]; in ComputeReshapePassthroughDimPairs()398 return absl::StrCat(value.result_dim, "->", in ComputeReshapePassthroughDimPairs()409 return lhs.result_dim < rhs.result_dim; in ComputeReshapePassthroughDimPairs()440 return it->result_dim; in MapPassthroughOperandDimToResultDim()[all …]
35 const Index result_dim = input_dim - kernel_dim + 1; in IndexMapper() local36 dimensions[index] = result_dim; in IndexMapper()339 const Index result_dim = input_dim - kernel_dim + 1;340 m_dimensions[index] = result_dim;358 const Index result_dim = input_dim - kernel_dim + 1;359 m_dimensions[index] = result_dim;785 const Index result_dim = input_dim - kernel_dim + 1;786 m_dimensions[index] = result_dim;