Searched refs:dims_out (Results 1 – 2 of 2) sorted by relevance
94 std::vector<int>* dims_out = output_array->mutable_shape()->mutable_dims(); in ComputeBinaryOperatorOutputSize() local95 dims_out->clear(); in ComputeBinaryOperatorOutputSize()96 dims_out->reserve(rank_out); in ComputeBinaryOperatorOutputSize()110 dims_out->push_back(dim_y); in ComputeBinaryOperatorOutputSize()114 dims_out->push_back(dim_x); in ComputeBinaryOperatorOutputSize()119 dims_out->push_back(dim_x); in ComputeBinaryOperatorOutputSize()
123 const std::vector<int>& dims_out) { in DescribeSparseTensorDenseMatMul() argument132 DescribeArbitraryRankOutput(dims_out, DT_FLOAT, &op_context.op_info); in DescribeSparseTensorDenseMatMul()